Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
| 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 |
-
|
| 56 |
-
|
| 57 |
-
|
| 58 |
-
|
|
|
|
| 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)
|