Spaces:

vinni1484
/

text-keywords

Runtime error

vinni1484 commited on May 19, 2022

Commit

dbc09f8

1 Parent(s): ef0970f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from keybert import KeyBERT
 def keywords(text):
     model = KeyBERT("sentence-transformers/xlm-r-distilroberta-base-paraphrase-v1")
-    keywords = model.extract_keywords("The quick brown fox jumps over the lazy dog.")
     return keywords
 gr.Interface(keywords, "text", "text",title="Keyword Extractor").launch()

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