MrGanesh commited on
Commit
2af242a
·
1 Parent(s): 3357678

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -12,16 +12,16 @@ patseer_stopwords=['a' , 'abaft' , 'abafter' , 'abaftest' , 'ability' , 'able' ,
12
 
13
  #from sklearn.feature_extraction.text import CountVectorizer
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
  keywords=kw_model.extract_keywords(text,candidates, keyphrase_ngram_range=(1, 3), top_n=50,vectorizer=KeyphraseCountVectorizer(stop_words=patseer_stopwords))
23
 
24
  if keywords != []:
25
  st.info("Extracted keywords")
26
- keywords = pd.DataFrame(keywords, columns=["Keyword", "Score"])
27
  st.table(keywords)
 
12
 
13
  #from sklearn.feature_extraction.text import CountVectorizer
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
  keywords=kw_model.extract_keywords(text,candidates, keyphrase_ngram_range=(1, 3), top_n=50,vectorizer=KeyphraseCountVectorizer(stop_words=patseer_stopwords))
23
 
24
  if keywords != []:
25
  st.info("Extracted keywords")
26
+ keywords=pd.DataFrame(keywords, columns=["Keyword", "Score"])
27
  st.table(keywords)