Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ def load_data():
|
|
25 |
|
26 |
data = load_dataset("cmudrc/wave-energy", data_files=["data.zip"], split='train')
|
27 |
geometry = numpy.reshape(data['geometry'], (S*N, G*G*G))
|
28 |
-
curves = numpy.reshape(data['curves'], (S*N, D*F))
|
29 |
return None, None, S, N, D, F, G, curves, geometry
|
30 |
|
31 |
# Disable eager execution because its bad
|
|
|
25 |
|
26 |
data = load_dataset("cmudrc/wave-energy", data_files=["data.zip"], split='train')
|
27 |
geometry = numpy.reshape(data['geometry'], (S*N, G*G*G))
|
28 |
+
curves = numpy.reshape(data['curves'], (S*N, D*F))
|
29 |
return None, None, S, N, D, F, G, curves, geometry
|
30 |
|
31 |
# Disable eager execution because its bad
|