Update app.py

app.py

@@ -253,11 +253,9 @@ def create_cam_visualization_pil(image_pil, cam, alpha=0.6, vis_threshold=0.2):
     cam_alpha = (cam_norm >= vis_threshold).astype(np.float32) * alpha  # Alpha mask
 
     cam_rgba = np.dstack((cam_colored, cam_alpha))  # Shape: (H, W, 4)
-
-    cam_rgba = np.repeat(np.repeat(cam_rgba, 8, axis=0), 8, axis=1)
     
     # Resize CAM to match image
-    cam_resized = Image.fromarray((cam_rgba * 255).astype(np.uint8), mode="RGBA").resize((size, size), resample=Image.Resampling.BICUBIC)
+    cam_resized = Image.fromarray((cam_rgba * 255).astype(np.uint8), mode="RGBA").resize((216,216), resample=Image.Resampling.NEAREST).resize((size, size), resample=Image.Resampling.BICUBIC)
 
     cam_image = transforms.CenterCrop((h, w))(cam_resized)