MrGanesh commited on
Commit
791c2b9
·
1 Parent(s): 0a1a5af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,12 +19,12 @@ pa_stopwords=['a' , 'abaft' , 'abafter' , 'abaftest' , 'ability' , 'able' , 'abo
19
  #from sklearn.feature_extraction.text import CountVectorizer
20
  from keybert import KeyBERT
21
  kw_model = KeyBERT(model='AI-Growth-Lab/PatentSBERTa')
22
- kw_extractor = yake.KeywordExtractor(top=50)
23
  candidates = kw_extractor.extract_keywords(text)
24
  keyphrases = [candidate[0] for candidate in candidates]
25
  from keyphrase_vectorizers import KeyphraseCountVectorizer
26
  keywords=kw_model.extract_keywords(text,keyphrases,
27
- top_n=50,stop_words=pa_stopwords,keyphrase_ngram_range=(1, 3),vectorizer=KeyphraseCountVectorizer(),use_mmr=True)
28
 
29
 
30
  if keywords != []:
 
19
  #from sklearn.feature_extraction.text import CountVectorizer
20
  from keybert import KeyBERT
21
  kw_model = KeyBERT(model='AI-Growth-Lab/PatentSBERTa')
22
+ kw_extractor = yake.KeywordExtractor(top=50,stopwords=pa_stopwords)
23
  candidates = kw_extractor.extract_keywords(text)
24
  keyphrases = [candidate[0] for candidate in candidates]
25
  from keyphrase_vectorizers import KeyphraseCountVectorizer
26
  keywords=kw_model.extract_keywords(text,keyphrases,
27
+ top_n=50,keyphrase_ngram_range=(1, 3),vectorizer=KeyphraseCountVectorizer(stop_words=pa_stopwords),use_mmr=True)
28
 
29
 
30
  if keywords != []: