Update main.py

main.py

@@ -28,7 +28,7 @@ data.reset_index(inplace=True)
 # Create a FAISS index for fast similarity search
 index = faiss.IndexFlatL2(len(data["embedding"][0]))
 index.metric_type = faiss.METRIC_INNER_PRODUCT
-index.add(numpy.stack(faiss.normalize_L2(data["embedding"].to_list()), axis=0))
+index.add(numpy.stack(faiss.normalize_L2(data["embedding"]), axis=0))
 
 
 # Define the search function