Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -50,7 +50,7 @@ if st.button("Train Autoencoder"):
|
|
| 50 |
x_test = x_test.astype('float32') / 255.0
|
| 51 |
|
| 52 |
x_train = np.reshape(x_train, (-1, 784)) # Flatten to (None, 784)
|
| 53 |
-
x_test = np
|
| 54 |
|
| 55 |
# Save test set in session state
|
| 56 |
st.session_state.x_test = x_test
|
|
|
|
| 50 |
x_test = x_test.astype('float32') / 255.0
|
| 51 |
|
| 52 |
x_train = np.reshape(x_train, (-1, 784)) # Flatten to (None, 784)
|
| 53 |
+
x_test = np.reshape(x_test, (-1, 784))
|
| 54 |
|
| 55 |
# Save test set in session state
|
| 56 |
st.session_state.x_test = x_test
|