Jackie2235 commited on
Commit
145e0c0
·
1 Parent(s): 6b4b2b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -9,8 +9,9 @@ path='checkpoint-210000'
9
  hf_model = pipeline("feature-extraction",model=path,tokenizer='bert-base-uncased' )
10
  etsy_model = KeyBERT(model=hf_model)
11
 
 
12
  text = st.text_area("Enter title here for the summarization:")
13
 
14
  if st.button('Generate'):
15
- out = etsy_model(text)
16
  st.json(out)
 
9
  hf_model = pipeline("feature-extraction",model=path,tokenizer='bert-base-uncased' )
10
  etsy_model = KeyBERT(model=hf_model)
11
 
12
+
13
  text = st.text_area("Enter title here for the summarization:")
14
 
15
  if st.button('Generate'):
16
+ out = etsy_modele.extract_keywords(test, keyphrase_ngram_range=(1,3), stop_words='english')
17
  st.json(out)