Update app.py

app.py

@@ -23,13 +23,10 @@ embeddings_array = np.array(data["Abstract_Embeddings"])
 # Check the shape of the embeddings
 print(embeddings_array.shape)
 
-# Reshape the embeddings if necessary
-if len(embeddings_array.shape) == 1:
-    embeddings_array = embeddings_array.reshape(-1, 1)
-
-# Create a new Dataset with the modified data
 new_data = data.remove_columns("Abstract_Embeddings")
 embeddings_array = np.array(data["Abstract_Embeddings"])
+if len(embeddings_array.shape) == 1:
+    embeddings_array = embeddings_array.reshape(-1, 1)
 new_data = new_data.add_column("Abstract_Embeddings", embeddings_array.tolist())
 
 # Convert the column to a numpy array