MrGanesh commited on
Commit
324fe53
·
1 Parent(s): b0247d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -47,12 +47,12 @@ button = st.button("Extract Keywords")
47
  #keyphrases = [candidate[0] for candidate in candidates]
48
  #kw_model = KeyBERT(model="google/bigbird-pegasus-large-bigpatent")
49
 
50
- print(predict_fn(text, model))
51
 
52
  #if keywords != []:
53
  #keywords = model.extract_keywords(text_input,keyphrases, keyphrase_ngram_range=(1, 3),
54
  #top_n=50,stop_words='english',vectorizer=KeyphraseCountVectorizer())
55
- #if keywords != []:
56
- #st.info("Extracted keywords")
57
- #keywords = pd.DataFrame(keywords, columns=["Keyword", "Score"])
58
- #st.table(keywords)
 
47
  #keyphrases = [candidate[0] for candidate in candidates]
48
  #kw_model = KeyBERT(model="google/bigbird-pegasus-large-bigpatent")
49
 
50
+ keywords=predict_fn(text, model)
51
 
52
  #if keywords != []:
53
  #keywords = model.extract_keywords(text_input,keyphrases, keyphrase_ngram_range=(1, 3),
54
  #top_n=50,stop_words='english',vectorizer=KeyphraseCountVectorizer())
55
+ if keywords != []:
56
+ st.info("Extracted keywords")
57
+ keywords = pd.DataFrame(keywords, columns=["Keyword", "Score"])
58
+ st.table(keywords)