Update app.py

app.py

@@ -6,7 +6,7 @@ from keyphrase_vectorizers import KeyphraseCountVectorizer
 
 @st.cache(allow_output_mutation=True, suppress_st_warning=True, show_spinner=True)
 def load_model():
-  model = KeyBERT("google/bigbird-pegasus-large-bigpatent")
+  model = KeyBERT("AI-Growth-Lab/PatentSBERTa")
   return model
   
 model = load_model()
@@ -37,10 +37,10 @@ button = st.button("Extract Keywords")
 kw_extractor = yake.KeywordExtractor(top=50)
 candidates = kw_extractor.extract_keywords(text_input)
 keyphrases = [candidate[0] for candidate in candidates]
-kw_model = KeyBERT(model="google/bigbird-pegasus-large-bigpatent")
+#kw_model = KeyBERT(model="google/bigbird-pegasus-large-bigpatent")
 
 #if keywords != []:
-keywords = kw_model.extract_keywords(text_input,keyphrases, keyphrase_ngram_range=(1, 3), 
+keywords = model.extract_keywords(text_input,keyphrases, keyphrase_ngram_range=(1, 3), 
                        top_n=50,stop_words='english',vectorizer=KeyphraseCountVectorizer())
 if keywords != []:
     st.info("Extracted keywords")