Update app.py

app.py

@@ -33,7 +33,7 @@ data = data.map(convert_and_ensure_2d_embeddings)
 
 # Flatten embeddings if they are nested 2D arrays
 def flatten_embeddings(example):
-    embedding = example['embedding']
+    embedding = np.array(example['embedding'], dtype=np.float32)
     if embedding.ndim == 2 and embedding.shape[0] == 1:
         embedding = embedding.flatten()
     return {'embedding': embedding}
@@ -44,7 +44,7 @@ data = data.map(flatten_embeddings)
 embeddings = np.array(data['embedding'].tolist(), dtype=np.float32)
 
 # Add FAISS index
-data.add_faiss_index_from_external_arrays(embeddings)
+data.add_faiss_index_from_external_arrays("embedding", embeddings)
 
 model_id = "meta-llama/Meta-Llama-3-8B-Instruct"