Update app.py

app.py

@@ -52,8 +52,18 @@ button = st.button("Extract Keywords")
 #candidates = kw_extractor.extract_keywords(text_input)
 #keyphrases = [candidate[0] for candidate in candidates]
 #kw_model = KeyBERT(model="google/bigbird-pegasus-large-bigpatent")
+import yake
+
+kw_extractor = yake.KeywordExtractor(top=50, stopwords=pa_stopwords)
+candidates = kw_extractor.extract_keywords(text)
+candidates = [candidate[0] for candidate in candidates]
+from keyphrase_vectorizers import KeyphraseCountVectorizer
+keywords=model.extract_keywords(text,candidates, keyphrase_ngram_range=(1, 3), top_n=50)
+
+
+
 
-keywords=predict_fn(text, model)
+#keywords=predict_fn(text, model)
 
 #if keywords != []:
 #keywords = model.extract_keywords(text_input,keyphrases, keyphrase_ngram_range=(1, 3),