Update app.py

app.py

@@ -246,7 +246,7 @@ def create_cam_visualization_pil(image_pil, cam, alpha=0.6, vis_threshold=0.2):
     cam_alpha = (cam >= vis_threshold).astype(np.float32) * alpha  # Alpha mask
     cam_rgba = np.dstack((cam_rgb, cam_alpha))  # Shape: (H, W, 4)
     
-    # Resize CAM to match image
+    # Coarse upscale for CAM output -- keeps "blocky" effect that is truer to what is measured
     cam_pil = Image.fromarray((cam_rgba * 255).astype(np.uint8), mode="RGBA")
     cam_pil = cam_pil.resize((216,216), resample=Image.Resampling.NEAREST)
 
@@ -255,7 +255,7 @@ def create_cam_visualization_pil(image_pil, cam, alpha=0.6, vis_threshold=0.2):
     cam_pil = transforms.CenterCrop((h, w))(cam_pil)
 
     # Composite over original
-    composite = Image.alpha_composite(image_pil, cam_image)
+    composite = Image.alpha_composite(image_pil, cam_pil)
 
     return composite