Update app.py

app.py

@@ -47,8 +47,9 @@ data = dataset
 d = 384  # vectors dimension
 m = 32  # hnsw parameter. Higher is more accurate but takes more time to index (default is 32, 128 should be ok)
 #index = faiss.IndexHNSWFlat(d, m)
-index =  faiss.IndexFlatL2(embedding_dim)
-data.add_faiss_index(embeddings.shape[1], custom_index=index) 
+#index =  faiss.IndexFlatL2(embedding_dim)
+index =  faiss.Index(embedding_dim)
+data.add_faiss_index("embeddings", custom_index=index) 
 #data.add_faiss_index("embeddings")
 # adds an index column for the embeddings