Update README.md

README.md

@@ -39,7 +39,7 @@ Keyphrase extraction is a technique in text analysis where you extract the impor
 
 ## 📓 Model Description
 This model is a fine-tuned KeyBART model on the Inspec dataset.
-KeyBART focuses on learning a better representation of keyphrases in a generative setting. It produces the keyphrases associated with the input. This is accomplished by predicting the original input based on a changed input. The input is changed by token masking, keyphrase masking and keyphrase replacement. [10] This model can already be used without any fine-tuning, but can be fine-tuned if needed.
+KeyBART focuses on learning a better representation of keyphrases in a generative setting. It produces the keyphrases associated with the input. This is accomplished by predicting the original input based on a changed input. The input is changed by token masking, keyphrase masking and keyphrase replacement. This model can already be used without any fine-tuning, but can be fine-tuned if needed.
 You can find more information about the architecture in this paper: https://arxiv.org/abs/2112.08547.
 
 Kulkarni, Mayank, Debanjan Mahata, Ravneet Arora, and Rajarshi Bhowmik. "Learning Rich Representation of Keyphrases from Text." arXiv preprint arXiv:2112.08547 (2021).
@@ -59,7 +59,6 @@ from transformers import (
     AutoModelForSeq2SeqLM,
     AutoTokenizer,
 )
-import numpy as np
 
 
 class KeyphraseGenerationPipeline(Text2TextGenerationPipeline):
@@ -76,7 +75,8 @@ class KeyphraseGenerationPipeline(Text2TextGenerationPipeline):
         results = super().postprocess(
             model_outputs=model_outputs
         )
-        return [[keyphrase.strip() for keyphrase in result.get("generated_text").split(self.keyphrase_sep_token)] for result in results]
+        return [[keyphrase.strip() for keyphrase in result.get("generated_text").split(self.keyphrase_sep_token) if keyphrase != ""] for result in results]
+
 ```
 
 ```python
@@ -88,10 +88,11 @@ generator = KeyphraseGenerationPipeline(model=model_name)
 text = """
 Keyphrase extraction is a technique in text analysis where you extract the important keyphrases from a text. 
 Since this is a time-consuming process, Artificial Intelligence is used to automate it. 
-Currently, classical machine learning methods, that use statistics and linguistics, are widely used for the extraction process. 
-The fact that these methods have been widely used in the community has the advantage that there are many easy-to-use libraries. 
-Now with the recent innovations in deep learning methods (such as recurrent neural networks and transformers, GANS, …), keyphrase extraction can be improved. 
-These new methods also focus on the semantics and context of a document, which is quite an improvement.
+Currently, classical machine learning methods, that use statistics and linguistics, 
+are widely used for the extraction process. The fact that these methods have been widely used in the community 
+has the advantage that there are many easy-to-use libraries. Now with the recent innovations in NLP, 
+transformers can be used to improve keyphrase extraction. Transformers also focus on the semantics 
+and context of a document, which is quite an improvement.
 """.replace(
     "\n", ""
 )
@@ -210,4 +211,4 @@ The model achieves the following results on the Inspec test set:
 For more information on the evaluation process, you can take a look at the keyphrase extraction evaluation notebook.
 
 ## 🚨 Issues
-Please feel free to contact Thomas De Decker for any problems with this model.
+Please feel free to start discussions in the Community Tab.