Update app.py

app.py

@@ -17,8 +17,8 @@ def load_data():
 
   from datasets import load_dataset
 
-  curves = load_dataset("cmudrc/wave-energy", data_files="curves.zip", split='train').to_pandas().drop(labels="index")
-  geometry = load_dataset("cmudrc/wave-energy", data_files="geometry.zip", split='train').to_pandas().drop(labels="index")
+  curves = load_dataset("cmudrc/wave-energy", data_files="curves.zip", split='train').to_pandas().drop(labels="index", axis=1)
+  geometry = load_dataset("cmudrc/wave-energy", data_files="geometry.zip", split='train').to_pandas().drop(labels="index", axis=1)
 
   # Open all the files we downloaded at the beginning and take out hte good bits
   # curves = data.iloc[:, [i for i in range(1, 3*64+1)]]