изменение весов

app.py

@@ -20,8 +20,9 @@ input_search = st.text_input('Search')
 tfidf_slider = st.sidebar.slider("tf_idf_description", 0.0, 1.0, 0.9)
 tf_idf_name = st.sidebar.slider("tf_idf_name", 0.0, 1.0, 0.66)/100
 tf_idf_actors = st.sidebar.slider("tf_idf_actors", 0.0, 1.0, 0.9)
-bert_weight = st.sidebar.slider("bert_weight", 0.0, 1.0, 0.9)
+bert_weight = st.sidebar.slider("bert_weight", 0.0, 1.0, 0.5)
 
+show_num = st.sidebar.slider("show_num", 1, 100, 100)
 
 data = np.load('data/embeddings_bert.npy')
 
@@ -97,7 +98,8 @@ def predict_rating(input_search,tfidf_slider,tf_idf_name,tf_idf_actors,bert_weig
        +similarity_actors*tf_idf_actors
        ).reshape(-1)  
     st.session_state["pred"]=y
-    return top_indices(y, 10,upsc=False)
+
+    return top_indices(y, show_num,upsc=False)