Jackie2235 commited on
Commit
7be320a
·
1 Parent(s): e5a04ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -13,7 +13,12 @@ etsy_model = KeyBERT(model=hf_model)
13
  text = st.text_area("Enter title here for the summarization:")
14
 
15
  if st.button('Generate'):
16
- out = etsy_model.extract_keywords(text, keyphrase_ngram_range=(1,3), stop_words='english')
17
  if len(out[0][0]) < 20:
18
- out = etsy_model.extract_keywords(text, keyphrase_ngram_range=(1,4), stop_words='english')
19
- st.json(out)
 
 
 
 
 
 
13
  text = st.text_area("Enter title here for the summarization:")
14
 
15
  if st.button('Generate'):
16
+ out = etsy_model.extract_keywords(text, keyphrase_ngram_range=(1,3), stop_words='english', top_n=20)
17
  if len(out[0][0]) < 20:
18
+ out = etsy_model.extract_keywords(text, keyphrase_ngram_range=(1,4), stop_words='english', top_n=20)
19
+
20
+ if len(out)==20:
21
+ show=[out[0],out[4],out[9],out[14],out[19]]
22
+ else:
23
+ show=out
24
+ st.json(show)