Update app.py

app.py

@@ -80,7 +80,8 @@ def inference(input_img, see_misclassified,num_misclassified_imgs,see_gradcam,nu
     #   input_img = org_img
 
     # else: # nothing chosen - misclassified or gradcam
-    #   org_img = input_img
+    
+    org_img = input_img
 
     # model inference
     transform = transforms.ToTensor()