MrGanesh commited on
Commit
016223f
·
1 Parent(s): b9b0b35

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -39,10 +39,13 @@ button = st.button("Extract Keywords")
39
  kw_extractor = yake.KeywordExtractor(top=50)
40
  candidates = kw_extractor.extract_keywords(text_input)
41
  keyphrases = [candidate[0] for candidate in candidates]
42
- kw_model = KeyBERT(model=model)
43
 
44
  #if keywords != []:
45
  st.info("Extracted keywords")
46
- keywords = kw_model.extract_keywords(text_input,candidates, keyphrase_ngram_range=(1, 3),
47
  top_n=50,stop_words='english',vectorizer=KeyphraseCountVectorizer())
48
- st.table(keywords)
 
 
 
 
39
  kw_extractor = yake.KeywordExtractor(top=50)
40
  candidates = kw_extractor.extract_keywords(text_input)
41
  keyphrases = [candidate[0] for candidate in candidates]
42
+ #kw_model = KeyBERT(model=model)
43
 
44
  #if keywords != []:
45
  st.info("Extracted keywords")
46
+ keywords = model.extract_keywords(text_input,candidates, keyphrase_ngram_range=(1, 3),
47
  top_n=50,stop_words='english',vectorizer=KeyphraseCountVectorizer())
48
+ if keywords != []:
49
+ st.info("Extracted keywords")
50
+ keywords = pd.DataFrame(keywords, columns=["Keyword", "Score"])
51
+ st.table(keywords)