MrGanesh commited on
Commit
5419ded
·
1 Parent(s): f05cd4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -16,9 +16,9 @@ from keybert import KeyBERT
16
  kw_model=KeyBERT(model='AI-Growth-Lab/PatentSBERTa')
17
  import yake
18
  #vectorizer = CountVectorizer(ngram_range=(1, 3), stop_words=patseer_stopwords)
19
- #kw_extractor=yake.KeywordExtractor(top=50)
20
- #candidates=kw_extractor.extract_keywords(text)
21
- #candidates=[candidate[0] for candidate in candidates]
22
  from keyphrase_vectorizers import KeyphraseCountVectorizer
23
  vectorizer=KeyphraseCountVectorizer(stop_words=patseer_stopwords)
24
  keywords=kw_model.extract_keywords(text,keyphrase_ngram_range=(1, 3),top_n=50,vectorizer=KeyphraseCountVectorizer())
 
16
  kw_model=KeyBERT(model='AI-Growth-Lab/PatentSBERTa')
17
  import yake
18
  #vectorizer = CountVectorizer(ngram_range=(1, 3), stop_words=patseer_stopwords)
19
+ kw_extractor=yake.KeywordExtractor(top=50)
20
+ candidates=kw_extractor.extract_keywords(text)
21
+ candidates=[candidate[0] for candidate in candidates]
22
  from keyphrase_vectorizers import KeyphraseCountVectorizer
23
  vectorizer=KeyphraseCountVectorizer(stop_words=patseer_stopwords)
24
  keywords=kw_model.extract_keywords(text,keyphrase_ngram_range=(1, 3),top_n=50,vectorizer=KeyphraseCountVectorizer())