MrGanesh commited on
Commit
70f3990
·
1 Parent(s): d3017f6

Update app.py

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