Update app.py

app.py

@@ -99,7 +99,8 @@ epsilon = 0.20  # Tweak epsilon to change the intensity of perturbation
 adversarial_examples = []
 
 for i in range(10):
-    img = tf.convert_to_tensor(mnist_examples[i][0].reshape((1, 28, 28, 1)), dtype=tf.float32)
+    img = preprocess_image(mnist_examples[i][0])
+    img = tf.convert_to_tensor(img, dtype=tf.float32)
     label = tf.reshape(tf.convert_to_tensor([i]), [1, 1])
     perturbations = create_adversarial_pattern(img, label)
     adv_x = fgsm_attack(img, epsilon, perturbations)