Update app.py

app.py

@@ -41,7 +41,7 @@ def flatten_embeddings(example):
 data = data.map(flatten_embeddings)
 
 # Ensure embeddings are in the correct shape for FAISS
-embeddings = np.array(data['embedding'].tolist(), dtype=np.float32)
+embeddings = np.array([example['embedding'] for example in data], dtype=np.float32)
 
 # Add FAISS index
 data.add_faiss_index_from_external_arrays("embedding", embeddings)