Update app.py

app.py

@@ -17,15 +17,15 @@ dataset = load_dataset("Yoxas/statistical_literacyv2")
 data = dataset["train"]
 
 # Check the structure of embeddings
-#example_embedding = data[0]['Abstract']
-#print(f"Example embedding shape: {np.array(example_embedding).shape}")
+example_embedding = data[0]['embedding']
+print(f"Example embedding shape: {np.array(example_embedding).shape}")
 
 # Ensure embeddings are 2-dimensional
-#def ensure_2d_embeddings(embeddings):
-    #return [np.atleast_2d(embedding) for embedding in embeddings]
+def ensure_2d_embeddings(embeddings):
+    return [np.atleast_2d(embedding) for embedding in embeddings]
 
 # Apply the function to ensure embeddings are 2-dimensional
-#data = data.map(lambda example: {'Abstract_Embeddings': ensure_2d_embeddings(example['Abstract_Embeddings'])})
+data = data.map(lambda example: {'embedding': ensure_2d_embeddings(example['embedding'])})
 
 data = data.add_faiss_index("embedding")