fixed issue in train bool

app.py

@@ -231,9 +231,9 @@ optimizer = optim.SGD(network.parameters(), lr=learning_rate,
                       momentum=momentum)        
 
 
-def train_and_test(train=True):
+def train_and_test(train_model=True):
 
-    if train:
+    if train_model:
         # Train for one epoch and test
         train_dataset = MNISTAdversarial_Dataset('./data_mnist',TRAIN_TRANSFORM)