Hem345 commited on
Commit
0292a29
·
verified ·
1 Parent(s): 7cc9dcf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 reshape(x_test, (-1, 784))
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