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)
|