Update app.py

app.py

@@ -20,7 +20,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']]
+  # round_geometry = [numpy.array(x) for x in data['geometry']]
+  round_geometry = numpy.array(data['geometry'])
   geometry = numpy.reshape(data['geometry'], (5000, 32*32*32))
   curves = numpy.reshape(data['curves'], (5000, 3*64))
   # curves = load_dataset("cmudrc/wave-energy", data_files="curves.zip", split='train').to_pandas().drop(labels="index", axis=1)