Update README.md

README.md

@@ -71,7 +71,7 @@ class KeyphraseGenerationPipeline(Text2TextGenerationPipeline):
         results = super().postprocess(
             model_outputs=model_outputs
         )
-        return [result.strip() for result in results[0].get("generated_text").split(self.keyphrase_sep_token)]
+        return [[keyphrase.strip() for keyphrase in result.get("generated_text").split(self.keyphrase_sep_token)] for result in results]
 ```
 
 ```python
@@ -99,7 +99,7 @@ print(keyphrases)
 
 ```
 # Output
-['keyphrase extraction', 'text analysis', 'artificial intelligence', 'classical machine learning', 'statistics']
+[['keyphrase extraction', 'text analysis', 'artificial intelligence', 'classical machine learning', 'statistics']]
 ```
 
 ## 📚 Training Dataset