MrGanesh commited on
Commit
74c6ae0
·
1 Parent(s): 1a6f3a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import streamlit as st
3
  from keybert import KeyBERT
4
  import yake
5
  from keyphrase_vectorizers import KeyphraseCountVectorizer
6
-
7
  @st.cache(allow_output_mutation=True, suppress_st_warning=True, show_spinner=True)
8
  def load_model():
9
  model = KeyBERT("AI-Growth-Lab/PatentSBERTa")
@@ -24,7 +24,7 @@ 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))
28
 
29
 
30
  if keywords != []:
 
3
  from keybert import KeyBERT
4
  import yake
5
  from keyphrase_vectorizers import KeyphraseCountVectorizer
6
+ import fastcountvectorizer
7
  @st.cache(allow_output_mutation=True, suppress_st_warning=True, show_spinner=True)
8
  def load_model():
9
  model = KeyBERT("AI-Growth-Lab/PatentSBERTa")
 
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=fastcountvectorizer())
28
 
29
 
30
  if keywords != []: