Update app.py

app.py

@@ -17,12 +17,18 @@ dataset = load_dataset("Yoxas/statistical_literacyv2")
 
 data = dataset["train"]
 
+# Convert the list to a numpy array
+embeddings_array = np.array(data["Abstract_Embeddings"])
+
 # Check the shape of the embeddings
-print(data["Abstract_Embeddings"].shape)
+print(embeddings_array.shape)
 
 # Reshape the embeddings if necessary
-if len(data["Abstract_Embeddings"].shape) == 1:
-    data["Abstract_Embeddings"] = data["Abstract_Embeddings"].reshape(-1, 1)
+if len(embeddings_array.shape) == 1:
+    embeddings_array = embeddings_array.reshape(-1, 1)
+
+# Replace the list with the numpy array
+data["Abstract_Embeddings"] = embeddings_array
 
 data = data.add_faiss_index("Abstract_Embeddings")