HaggiVaggi commited on
Commit
1adb805
1 Parent(s): ff6cf2b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -13
app.py CHANGED
@@ -67,27 +67,27 @@ if page == "какая-то еще":
67
  # embeddings = pd.read_pickle('embeddings.pkl')
68
 
69
  def main():
70
- st.title("Система поиска фильмов")
71
 
72
  # Пользовательский ввод
73
- user_input = st.text_input("Введите описание фильма:")
74
 
75
- if st.button("Искать🔍🎦')"):
76
- if user_input:
77
  # Векторизация введенного запроса
78
- input_embedding = encode_description(user_input)
79
 
80
  # Поиск с использованием Faiss
81
- _, sorted_indices = index.search(input_embedding.reshape(1, -1), 5) # Изменил на 5
82
 
83
  # Используйте индексы для извлечения строк из DataFrame
84
- recs = df.iloc[sorted_indices[0]].reset_index(drop=True)
85
- recs.index = recs.index + 1
86
 
87
  # Вывод рекомендованных фильмов с изображениями
88
- st.write("Рекомендованные фильмы:")
89
- for i in range(5): # Изменил на 5
90
- st.image(recs['image_url'].iloc[i], caption=recs[['movie_title', 'description']].iloc[i], use_column_width=True)
91
 
92
- if __name__ == "__main__":
93
- main()
 
67
  # embeddings = pd.read_pickle('embeddings.pkl')
68
 
69
  def main():
70
+ st.title("Система поиска фильмов")
71
 
72
  # Пользовательский ввод
73
+ user_input = st.text_input("Введите описание фильма:")
74
 
75
+ if st.button("Искать🔍🎦')"):
76
+ if user_input:
77
  # Векторизация введенного запроса
78
+ input_embedding = encode_description(user_input)
79
 
80
  # Поиск с использованием Faiss
81
+ _, sorted_indices = index.search(input_embedding.reshape(1, -1), 5) # Изменил на 5
82
 
83
  # Используйте индексы для извлечения строк из DataFrame
84
+ recs = df.iloc[sorted_indices[0]].reset_index(drop=True)
85
+ recs.index = recs.index + 1
86
 
87
  # Вывод рекомендованных фильмов с изображениями
88
+ st.write("Рекомендованные фильмы:")
89
+ for i in range(5): # Изменил на 5
90
+ st.image(recs['image_url'].iloc[i], caption=recs[['movie_title', 'description']].iloc[i], use_column_width=True)
91
 
92
+ if __name__ == "__main__":
93
+ main()