Update app.py

app.py

@@ -1,6 +1,8 @@
 import pandas as pd
 import streamlit as st
 from keybert import KeyBERT
+import yake
+from keyphrase_vectorizers import KeyphraseCountVectorizer
 
 @st.cache(allow_output_mutation=True, suppress_st_warning=True, show_spinner=True)
 def load_model():
@@ -12,12 +14,12 @@ model = load_model()
 placeholder = st.empty()
 text_input = placeholder.text_area("Paste or write text", height=300)
 
-top_n = st.sidebar.slider("Select a number of keywords", 1, 10, 5, 1)
+top_n = st.sidebar.slider("Select a number of keywords", 1, 10, 50,20)
 min_ngram = st.sidebar.number_input("Minimum number of words in each keyword", 1, 5, 1, 1)
 max_ngram = st.sidebar.number_input("Maximum number of words in each keyword", min_ngram, 5, 3, step=1)
 st.sidebar.code(f"ngram_range=({min_ngram}, {max_ngram})")
 
-params = {"docs": text_input, "top_n": top_n, "keyphrase_ngram_range": (min_ngram, max_ngram), "stop_words": 'english'}
+params = {"docs": text_input, "top_n": top_n, "keyphrase_ngram_range": (min_ngram, max_ngram), "stop_words": 'english',vectorizer=KeyphraseCountVectorizer()}
 
 add_diversity = st.sidebar.checkbox("Adjust diversity of keywords")