MrGanesh commited on
Commit
15bb13b
·
1 Parent(s): 791c2b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -21,11 +21,9 @@ 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 != []:
31
  st.info("Extracted keywords")
 
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
+ candidates = [candidate[0] for candidate in candidates]
25
  from keyphrase_vectorizers import KeyphraseCountVectorizer
26
+ keywords=model.extract_keywords(text,candidates,top_n=50,keyphrase_ngram_range=(1,3),vectorizer=KeyphraseCountVectorizer(stop_words=pa_stopwords),use_mmr=True)
 
 
27
 
28
  if keywords != []:
29
  st.info("Extracted keywords")