Spaces:

vinni1484
/

text-keywords

Runtime error

vinni1484 commited on May 19, 2022

Commit

6d46237

1 Parent(s): aa697df

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)
     return keywords
-gr.Interface(keywords, "text", "text",title="Keyword Extractor",top_n=10).launch()

 model = KeyBERT("sentence-transformers/xlm-r-distilroberta-base-paraphrase-v1")
 def keywords(text):
+    keywords = model.extract_keywords(text,top_n=10)
     return keywords
+gr.Interface(keywords, "text", "text",title="Keyword Extractor").launch()