MrGanesh commited on
Commit
5b45828
·
1 Parent(s): 6a4af56

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,8 +6,8 @@ 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("google/bigbird-pegasus-large-bigpatent")
10
- #return model
11
 
12
  model = load_model()
13
  st.title("Patent Text Extractor")
 
6
 
7
  @st.cache(allow_output_mutation=True, suppress_st_warning=True, show_spinner=True)
8
  def load_model():
9
+ model = KeyBERT("google/bigbird-pegasus-large-bigpatent")
10
+ return model
11
 
12
  model = load_model()
13
  st.title("Patent Text Extractor")