Update app.py

app.py

@@ -42,18 +42,23 @@ nlist = 150
 quantizer = faiss.IndexFlatIP(dimension)
 
 
-index = faiss.IndexIVFFlat(quantizer, dimension, nlist, faiss.METRIC_L1)
+index = faiss.IndexIVFFlat(quantizer, dimension, nlist, faiss.METRIC_INNER_PRODUCT)
 index.train(embeddings)
 # index = faiss.IndexFlatIP(dimension)
 
 index.add(embeddings)
 
 query = [st.text_area('Введите описание сериала')]
+
+
 button = st.button('Вывести результаты')
 if button:
     if query:
         query_embedding = model.encode(query).astype(np.float32)
-    
+        # Две строки ниже можно будет убрать
+        query_embedding = np.array(query_embedding, dtype=np.float32).reshape(1, -1)
+        faiss.normalize_L2(query_embedding)
+        
         k = st.slider('Сколько сериалов рекомендовать?',
                       min_value=1, max_value=10, value=3, step=1)
         distances, indices = index.search(query_embedding, k)