Update app.py

app.py

@@ -18,7 +18,7 @@ def load_data():
   from datasets import load_dataset
 
 
-  data = load_dataset("cmudrc/wave-energy", data_files="data.zip")
+  data = load_dataset("cmudrc/wave-energy", data_files=["data.zip"], split='train')
   curvey_curves = [numpy.array(x)/1000000 for x in data['curves']]
   round_geometry = [numpy.array(x) for x in data['geometry']]
   geometry = [x.flatten() for x in round_geometry]