Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -19,12 +19,12 @@ pa_stopwords=['a' , 'abaft' , 'abafter' , 'abaftest' , 'ability' , 'able' , 'abo
|
|
19 |
#from sklearn.feature_extraction.text import CountVectorizer
|
20 |
from keybert import KeyBERT
|
21 |
kw_model = KeyBERT(model='AI-Growth-Lab/PatentSBERTa')
|
22 |
-
kw_extractor = yake.KeywordExtractor(top=50)
|
23 |
candidates = kw_extractor.extract_keywords(text)
|
24 |
keyphrases = [candidate[0] for candidate in candidates]
|
25 |
from keyphrase_vectorizers import KeyphraseCountVectorizer
|
26 |
keywords=kw_model.extract_keywords(text,keyphrases,
|
27 |
-
top_n=50,
|
28 |
|
29 |
|
30 |
if keywords != []:
|
|
|
19 |
#from sklearn.feature_extraction.text import CountVectorizer
|
20 |
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 |
keyphrases = [candidate[0] for candidate in candidates]
|
25 |
from keyphrase_vectorizers import KeyphraseCountVectorizer
|
26 |
keywords=kw_model.extract_keywords(text,keyphrases,
|
27 |
+
top_n=50,keyphrase_ngram_range=(1, 3),vectorizer=KeyphraseCountVectorizer(stop_words=pa_stopwords),use_mmr=True)
|
28 |
|
29 |
|
30 |
if keywords != []:
|