Update app.py

app.py

@@ -15,8 +15,6 @@ ST = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
 
 dataset = load_dataset("Yoxas/statistical_literacyv2")
 
-data = dataset["train"]
-
 # Convert the list to a numpy array
 embeddings_array = np.array(data["Abstract_Embeddings"])
 
@@ -34,8 +32,11 @@ new_data = new_data.add_column("Abstract_Embeddings", embeddings_array.tolist())
 # Convert the column to a numpy array
 new_data.set_format("numpy", columns=["Abstract_Embeddings"])
 
-new_data = new_data.add_faiss_index("Abstract_Embeddings")
+# Initialize the Faiss index
+new_data.add_faiss_index("Abstract_Embeddings")
 
+# Now you can use the Dataset with the Faiss index
+data = dataset["train"]
 
 model_id = "meta-llama/Meta-Llama-3-8B-Instruct"