VerVelVel commited on
Commit
9cf6a6a
·
1 Parent(s): 953c27c
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -64,13 +64,13 @@ def search_series(user_query, top_k, description_weight, actors_weight, genre_we
64
  # results['cosine_similarity'] = cosine_similarities
65
  return I[0], D[0]
66
 
67
- # Инициализация весов
68
- if 'original_description_weight' not in st.session_state:
69
- st.session_state['original_description_weight'] = 0.7
70
- if 'original_actors_weight' not in st.session_state:
71
- st.session_state['original_actors_weight'] = 0.15
72
- if 'original_genre_weight' not in st.session_state:
73
- st.session_state['original_genre_weight'] = 0.15
74
 
75
  # Функция для сброса весов к первоначальным значениям
76
  def reset_weights():
 
64
  # results['cosine_similarity'] = cosine_similarities
65
  return I[0], D[0]
66
 
67
+ # Инициализация весов в session_state
68
+ if 'description_weight' not in st.session_state:
69
+ st.session_state['description_weight'] = 0.7
70
+ if 'actors_weight' not in st.session_state:
71
+ st.session_state['actors_weight'] = 0.15
72
+ if 'genre_weight' not in st.session_state:
73
+ st.session_state['genre_weight'] = 0.15
74
 
75
  # Функция для сброса весов к первоначальным значениям
76
  def reset_weights():