MrGanesh commited on
Commit
0b2b0ad
·
1 Parent(s): 19080cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
- candidates = [candidate[0] for candidate in candidates]
47
  from keyphrase_vectorizers import KeyphraseCountVectorizer
48
- keywords=kw_model.extract_keywords(text,candidates,
49
- top_n=50,stop_words='english',vectorizer=KeyphraseCountVectorizer())
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)