Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -43,10 +43,10 @@ from keybert import KeyBERT
|
|
43 |
kw_model = KeyBERT(model='AI-Growth-Lab/PatentSBERTa')
|
44 |
kw_extractor = yake.KeywordExtractor(top=50)
|
45 |
candidates = kw_extractor.extract_keywords(text)
|
46 |
-
|
47 |
from keyphrase_vectorizers import KeyphraseCountVectorizer
|
48 |
-
keywords=kw_model.extract_keywords(text,
|
49 |
-
top_n=50,stop_words='english'
|
50 |
|
51 |
|
52 |
#keywords=predict_fn(text, model)
|
|
|
43 |
kw_model = KeyBERT(model='AI-Growth-Lab/PatentSBERTa')
|
44 |
kw_extractor = yake.KeywordExtractor(top=50)
|
45 |
candidates = kw_extractor.extract_keywords(text)
|
46 |
+
keyphrases = [candidate[0] for candidate in candidates]
|
47 |
from keyphrase_vectorizers import KeyphraseCountVectorizer
|
48 |
+
keywords=kw_model.extract_keywords(text,keyphrases,
|
49 |
+
top_n=50,stop_words='english')
|
50 |
|
51 |
|
52 |
#keywords=predict_fn(text, model)
|