Update app.py

app.py

@@ -17,10 +17,11 @@ def read_in_text(url):
 tmp_model = SentenceTransformer('valurank/MiniLM-L6-Keyword-Extraction')
 kw_extractor = KeyBERT(tmp_model)
 
-def get_keybert_results_with_vectorizer(text, number_of_results=30):
+def get_keybert_results_with_vectorizer(file, number_of_results=20):
     try:
+        text = read_in_text(file.name)
         keywords = kw_extractor.extract_keywords(text, vectorizer=KeyphraseCountVectorizer(), stop_words=None, top_n=number_of_results)
-        keywords = [i for i in keywords if i[1] >= 0.20]
+        keywords = [i for i in keywords if i[1] > 0.20]
 
         keybert_diversity_phrases = []
         for i, j in keywords:
@@ -32,7 +33,7 @@ def get_keybert_results_with_vectorizer(text, number_of_results=30):
     except Exception:
         return "Error"
         
-demo = gr.Interface(get_keybert_results_with_vectorizer, inputs=gr.Textbox(),
+demo = gr.Interface(get_keybert_results_with_vectorizer, inputs=gr.inputs.File(),
                     outputs=gr.outputs.Dataframe(),
                     title = "Keyword Extraction")