Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -39,10 +39,13 @@ button = st.button("Extract Keywords")
|
|
39 |
kw_extractor = yake.KeywordExtractor(top=50)
|
40 |
candidates = kw_extractor.extract_keywords(text_input)
|
41 |
keyphrases = [candidate[0] for candidate in candidates]
|
42 |
-
kw_model = KeyBERT(model=model)
|
43 |
|
44 |
#if keywords != []:
|
45 |
st.info("Extracted keywords")
|
46 |
-
keywords =
|
47 |
top_n=50,stop_words='english',vectorizer=KeyphraseCountVectorizer())
|
48 |
-
|
|
|
|
|
|
|
|
39 |
kw_extractor = yake.KeywordExtractor(top=50)
|
40 |
candidates = kw_extractor.extract_keywords(text_input)
|
41 |
keyphrases = [candidate[0] for candidate in candidates]
|
42 |
+
#kw_model = KeyBERT(model=model)
|
43 |
|
44 |
#if keywords != []:
|
45 |
st.info("Extracted keywords")
|
46 |
+
keywords = model.extract_keywords(text_input,candidates, keyphrase_ngram_range=(1, 3),
|
47 |
top_n=50,stop_words='english',vectorizer=KeyphraseCountVectorizer())
|
48 |
+
if keywords != []:
|
49 |
+
st.info("Extracted keywords")
|
50 |
+
keywords = pd.DataFrame(keywords, columns=["Keyword", "Score"])
|
51 |
+
st.table(keywords)
|