Update app.py

app.py

@@ -135,7 +135,7 @@ def fn(dataset, attack):
         pred1 = np.argmax(label1)
         if attack == "FGSM":
             image2 = fgsm_mnist(image1, label1, model_mnist, epsilon)
-        elif attack == "I_FGSM":
+        elif attack == "I-FGSM":
             image2 = iterative_fgsm_mnist(image1, label1, model_mnist, epsilon, alpha, niter)
         else:
             image2 = iterative_least_likely_fgsm_mnist(image1, model_mnist, epsilon, alpha, niter, nb_classes)
@@ -150,7 +150,7 @@ def fn(dataset, attack):
         pred1 = classes[np.argmax(label1)]
         if attack == "FGSM":
             image2 = fgsm_cifar10(image1, label1, model_cifar10, epsilon)
-        elif attack == "I_FGSM":
+        elif attack == "I-FGSM":
             image2 = iterative_fgsm_cifar10(image1, label1, model_cifar10, epsilon, alpha, niter)
         else:
             image2 = iterative_least_likely_fgsm_cifar10(image1, model_cifar10, epsilon, alpha, niter, nb_classes)