Update app.py

app.py

@@ -82,7 +82,7 @@ def gradio_callback(view_grad_cam, num_gradcam_images, view_misclassified, num_m
                     input_img, transparency = 0.5, target_layer_number = -1):
     
     confidence = inference_confidences(input_img, transparency = 0.5, target_layer_number = -1)
-    if view_grad_cam:
+    if view_grad_cam == "Yes":
         visualization = inference_visualization(input_img, transparency = 0.5, target_layer_number = -1)
         return confidence, visualization
     else: