Update app.py

app.py

@@ -21,8 +21,8 @@ def load_data():
   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]
-  curves = [x.flatten() for x in curvey_curves]
+  geometry = numpy.reshape(json_data['geometry'], (5000, 32*32*32))
+  curves = numpy.reshape(json_data['curves'], (5000, 3*64))
   # 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)