Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -14,8 +14,8 @@ def predict_fn(text, model):
|
|
14 |
kw_extractor = yake.KeywordExtractor(top=50)
|
15 |
candidates = kw_extractor.extract_keywords(text)
|
16 |
keyphrases = [candidate[0] for candidate in candidates]
|
17 |
-
kw_model = KeyBERT(model=kw_model)
|
18 |
-
keywords=
|
19 |
return keywords
|
20 |
|
21 |
st.title("Patent Text Extractor")
|
|
|
14 |
kw_extractor = yake.KeywordExtractor(top=50)
|
15 |
candidates = kw_extractor.extract_keywords(text)
|
16 |
keyphrases = [candidate[0] for candidate in candidates]
|
17 |
+
#kw_model = KeyBERT(model=kw_model)
|
18 |
+
keywords=model.extract_keywords(text, keyphrases, keyphrase_ngram_range=(1, 3), top_n=50)
|
19 |
return keywords
|
20 |
|
21 |
st.title("Patent Text Extractor")
|