Update app.py

app.py

@@ -199,6 +199,7 @@ def hook_backward(module, grad_in, grad_out):
     gradients['value'] = grad_out[0]
 
 def cam_inference(target_tag, threshold):
+    print(f"target_tag: {target_tag}")
     global input_image, sorted_tag_score, target_tag_index, gradients, activations
     img = input_image
     tensor = transform(img).unsqueeze(0)
@@ -260,9 +261,7 @@ def create_cam_visualization_pil(cam, alpha=0.6, vis_threshold=0.2):
     Returns:
         PIL.Image.Image with overlay
     """
-    if cam is None:
-        print("CAM is None, skipping visualization.")
-        return image_pil
+
     global input_image
     # Convert to RGB (in case RGBA or others)
     image_pil = input_image
@@ -336,7 +335,7 @@ with gr.Blocks(css=".output-class { display: none; }") as demo:
 
     label_box.select(
         fn=cam_inference,
-        inputs=[threshold_slider],
+        inputs=[label_box, threshold_slider],
         outputs=[image_input]
     )