Update app.py

app.py

@@ -7,6 +7,7 @@ import numpy as np
 
 
 
+
 st.title('Умный поиск фильмов 🔍🎦')
 
 with st.sidebar:
@@ -53,7 +54,7 @@ if page == "какая-то еще":
     tokenizer = AutoTokenizer.from_pretrained("DeepPavlov/rubert-base-cased-sentence")
     model = AutoModel.from_pretrained("DeepPavlov/rubert-base-cased-sentence")
         
-    df2 = pd.read_csv('data_with_embeddings.csv')   
+    df2 = pd.read_csv('data_with_adds.csv') 
     embeddings_array = np.load('embeddings.npy')
     index = faiss.read_index('desc_faiss_index.index')
     
@@ -81,7 +82,7 @@ if page == "какая-то еще":
                 _, sorted_indices = index.search(input_embedding.reshape(1, -1), 5)  # Изменил на 5
 
             # Используйте индексы для извлечения строк из DataFrame
-                recs = df.iloc[sorted_indices[0]].reset_index(drop=True)
+                recs = df2.iloc[sorted_indices[0]].reset_index(drop=True)
                 recs.index = recs.index + 1
 
             # Вывод рекомендованных фильмов с изображениями