Spaces:
Runtime error
Runtime error
Commit
·
145e0c0
1
Parent(s):
6b4b2b8
Update app.py
Browse files
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 =
|
| 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)
|