Spaces:
Sleeping
Sleeping
Commit
·
1adb805
1
Parent(s):
ff6cf2b
Update app.py
Browse files
app.py
CHANGED
@@ -67,27 +67,27 @@ if page == "какая-то еще":
|
|
67 |
# embeddings = pd.read_pickle('embeddings.pkl')
|
68 |
|
69 |
def main():
|
70 |
-
|
71 |
|
72 |
# Пользовательский ввод
|
73 |
-
|
74 |
|
75 |
-
|
76 |
-
|
77 |
# Векторизация введенного запроса
|
78 |
-
|
79 |
|
80 |
# Поиск с использованием Faiss
|
81 |
-
|
82 |
|
83 |
# Используйте индексы для извлечения строк из DataFrame
|
84 |
-
|
85 |
-
|
86 |
|
87 |
# Вывод рекомендованных фильмов с изображениями
|
88 |
-
|
89 |
-
|
90 |
-
|
91 |
|
92 |
-
if __name__ == "__main__":
|
93 |
-
|
|
|
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()
|