Update app.py

app.py

@@ -6,7 +6,7 @@ def process_images_gradcam(show_gradcam, gradcam_count, gradcam_layer, gradcam_o
             std=[4.0486, 4.1152, 3.8314])
         classes = ('plane', 'car', 'bird', 'cat', 'deer',
            'dog', 'frog', 'horse', 'ship', 'truck')
-        misclassified_data = get_misclassified_data(modelfin, "cuda", test_loader)
+        misclassified_data = get_misclassified_data(modelfin, "cpu", test_loader)
         if gradcam_layer=="1":
             images = display_gradcam_output(misclassified_data, classes, inv_normalize, modelfin, target_layers= [modelfin.model.layer1[-1]], targets=None, number_of_samples=gradcam_count, transparency=gradcam_opacity)
         if gradcam_layer=="2":
@@ -19,7 +19,7 @@ def process_images_gradcam(show_gradcam, gradcam_count, gradcam_layer, gradcam_o
     
 def process_images_misclass(show_misclassify, misclassify_count):
     if show_misclassify:
-        misclassified_data = get_misclassified_data(modelfin, "cuda", test_loader)    
+        misclassified_data = get_misclassified_data(modelfin, "cpu", test_loader)    
         image = display_cifar_misclassified_data(misclassified_data, classes, inv_normalize, number_of_samples=misclassify_count)    
         return image