Update app.py

app.py

@@ -19,14 +19,14 @@ data = dataset["train"]
 example_embedding = data[0]['embedding']
 print(f"Example embedding shape: {np.array(example_embedding).shape}")
 
-# Ensure embeddings are 2-dimensional
+# Ensure embeddings are 2-dimensional and of type float32
 def ensure_2d_embeddings(embeddings):
-    embeddings = np.array(embeddings)
+    embeddings = np.array(embeddings, dtype=np.float32)
     if embeddings.ndim == 1:
         embeddings = embeddings.reshape(1, -1)
     return embeddings
 
-# Apply the function to ensure embeddings are 2-dimensional
+# Apply the function to ensure embeddings are 2-dimensional and of type float32
 data = data.map(lambda example: {'embedding': ensure_2d_embeddings(example['embedding'])})
 
 data = data.add_faiss_index("embedding")