Update app.py

app.py

@@ -21,9 +21,10 @@ 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']]
-  round_geometry = numpy.array(data['geometry'])
+  # round_geometry = numpy.array(data['geometry'])
+  round_geometry = None
   geometry = numpy.reshape(data['geometry'], (5000, 32*32*32))
-  curves = numpy.reshape(data['curves'], (5000, 3*64))
+  curves = numpy.reshape(data['curves'], (5000, 3*64))/1000000
   # 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)