Update app.py

app.py

@@ -27,10 +27,11 @@ print(embeddings_array.shape)
 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
+# Create a new Dataset with the modified data
+new_data = data.remove_column("Abstract_Embeddings")
+new_data = new_data.add_column("Abstract_Embeddings", embeddings_array)
 
-data = data.add_faiss_index("Abstract_Embeddings")
+new_data = new_data.add_faiss_index("Abstract_Embeddings")
 
 
 model_id = "meta-llama/Meta-Llama-3-8B-Instruct"