Spaces:

vinni1484
/

text-keywords

Runtime error

vinni1484 commited on May 19, 2022

Commit

deb1ba4

1 Parent(s): e86c696

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from keybert import KeyBERT
 model = KeyBERT("sentence-transformers/xlm-r-distilroberta-base-paraphrase-v1")
 def keywords(text):
-    keywords = model.extract_keywords(text,top_n=10)
     keywords = dict(keywords)
     return keywords

 model = KeyBERT("sentence-transformers/xlm-r-distilroberta-base-paraphrase-v1")
 def keywords(text):
+    keywords = model.extract_keywords(text,keyphrase_ngram_range=(1,2),top_n=10)
     keywords = dict(keywords)
     return keywords