MrGanesh commited on
Commit
2a7557b
·
1 Parent(s): 7ea96d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=kw_model.extract_keywords(text, keyphrases, keyphrase_ngram_range=(1, 3), top_n=50)
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")