Update app.py

app.py

@@ -12,7 +12,7 @@ y = mnist.target.astype('int64')
 X /= 255.0
 
 
-device = 'cuda' if torch.cuda.is_available() else 'cpu'
+#device = 'cuda' if torch.cuda.is_available() else 'cpu'
 XCnn = X.reshape(-1, 1, 28, 28)
 XCnn_train, XCnn_test, y_train, y_test = train_test_split(XCnn, y, test_size=0.25, random_state=42)