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("AI-Growth-Lab/PatentSBERTa")
+  model = KeyBERT("google/bigbird-pegasus-large-bigpatent")
   return model
   
 model = load_model()
@@ -19,7 +19,7 @@ top_n = st.sidebar.slider("Select a number of keywords", 1, 10, 50,20)
 #max_ngram = st.sidebar.number_input("Maximum number of words in each keyword", 3)
 #st.sidebar.code(f"ngram_range=({min_ngram}, {max_ngram})")
 
-params = {"docs": text_input, "top_n": top_n, "keyphrase_ngram_range": (1, 3), "stop_words": 'english',"vectorizer":KeyphraseCountVectorizer()}
+params = {"docs": text_input, "top_n": top_n, "ngram_range": (1, 3), "stop_words": 'english',"vectorizer":KeyphraseCountVectorizer()}
 
 #add_diversity = st.sidebar.checkbox("Adjust diversity of keywords")