Update app.py

app.py

@@ -29,14 +29,14 @@ def load_data():
   F = 64
   G = 32
 
-  flattened_curves = curves.values / 1000000
-  curvey_curves = [c.reshape([3, 64]) for c in flattened_curves]
+  # flattened_curves = curves.values / 1000000
+  curvey_curves = [c.reshape([D, F])/1000000 for c in curves.values]
 
-  flattened_geometry = geometry.values
-  round_geometry = [g.reshape([32, 32, 32]) for g in flattened_geometry]
+  # flattened_geometry = geometry.values
+  round_geometry = [g.reshape([G, G, G]) for g in geometry.values]
 
   # Return good bits to user
-  return curvey_curves, round_geometry, S, N, D, F, G, flattened_curves, flattened_geometry
+  return curvey_curves, round_geometry, S, N, D, F, G, curves.values/1000000, geometry.values
 
 # Disable eager execution because its bad
 from tensorflow.python.framework.ops import disable_eager_execution