Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -21,11 +21,9 @@ from keybert import KeyBERT
|
|
21 |
kw_model = KeyBERT(model='AI-Growth-Lab/PatentSBERTa')
|
22 |
kw_extractor = yake.KeywordExtractor(top=50,stopwords=pa_stopwords)
|
23 |
candidates = kw_extractor.extract_keywords(text)
|
24 |
-
|
25 |
from keyphrase_vectorizers import KeyphraseCountVectorizer
|
26 |
-
keywords=
|
27 |
-
top_n=50,keyphrase_ngram_range=(1, 3),vectorizer=KeyphraseCountVectorizer(stop_words=pa_stopwords),use_mmr=True)
|
28 |
-
|
29 |
|
30 |
if keywords != []:
|
31 |
st.info("Extracted keywords")
|
|
|
21 |
kw_model = KeyBERT(model='AI-Growth-Lab/PatentSBERTa')
|
22 |
kw_extractor = yake.KeywordExtractor(top=50,stopwords=pa_stopwords)
|
23 |
candidates = kw_extractor.extract_keywords(text)
|
24 |
+
candidates = [candidate[0] for candidate in candidates]
|
25 |
from keyphrase_vectorizers import KeyphraseCountVectorizer
|
26 |
+
keywords=model.extract_keywords(text,candidates,top_n=50,keyphrase_ngram_range=(1,3),vectorizer=KeyphraseCountVectorizer(stop_words=pa_stopwords),use_mmr=True)
|
|
|
|
|
27 |
|
28 |
if keywords != []:
|
29 |
st.info("Extracted keywords")
|