MrGanesh commited on
Commit
7fef245
1 Parent(s): 8f4d396

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -14,14 +14,14 @@ patseer_stopwords=['a' , 'abaft' , 'abafter' , 'abaftest' , 'ability' , 'able' ,
14
  #from sklearn.feature_extraction.text import CountVectorizer
15
  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,candidates, keyphrase_ngram_range=(1, 3),top_n=50,vectorizer=KeyphraseCountVectorizer())
25
 
26
  if keywords != []:
27
  st.info("Extracted keywords")
 
14
  #from sklearn.feature_extraction.text import CountVectorizer
15
  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())
25
 
26
  if keywords != []:
27
  st.info("Extracted keywords")