Update main.py

main.py

@@ -27,6 +27,7 @@ index = faiss.IndexFlatL2(len(data["embedding"][0]))
 index.metric_type = faiss.METRIC_INNER_PRODUCT
 vectors = numpy.stack(data["embedding"].tolist(), axis=0)
 faiss.normalize_L2(vectors)
+index.train(vectors)
 index.add(vectors)
 
 # Load the model for later use in embeddings