Update app.py

app.py

@@ -17,11 +17,12 @@ def load_data():
 
   from datasets import load_dataset
 
-  data = load_dataset("cmudrc/wave-energy", data_files="data.csv", split='train').to_pandas()
+  curves = load_dataset("cmudrc/wave-energy", data_files="curves.zip", split='train').to_pandas()
+  curves = load_dataset("cmudrc/wave-energy", data_files="geometry.zip", split='train').to_pandas()
 
   # 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)]]
-  geometry = data.iloc[:, [i for i in range(1 + 3*64, 1 + 3*64 + 32**3)]]
+  # curves = data.iloc[:, [i for i in range(1, 3*64+1)]]
+  # geometry = data.iloc[:, [i for i in range(1 + 3*64, 1 + 3*64 + 32**3)]]
   S = 5
   N = 1000
   D = 3