Update app.py

app.py

@@ -8,7 +8,7 @@ def demo1_derive_MNIST_train_test_data():
     X_train, X_test, y_train, y_test = X[:60000], X[60000:], y[:60000], y[60000:]
     y_train = y_train.astype(np.int) # convert to int
     y_test = y_test.astype(np.int) # convert to int
-    return X_train, X_test, y_train, y_test
+    return X_train, X_test, y_train, y_test 
 
 X_train, X_test, y_train, y_test = demo1_derive_MNIST_train_test_data()
 print("X_train.shape: ", X_train.shape)