Update app.py

app.py

@@ -156,9 +156,19 @@ def predict():
         
 
 
+        # ✅ Ensure cam is 2D
+        if cam.ndim == 3:
+            cam = cam[0]
+        
+        # ✅ Scale CAM and resize
+        cam = np.uint8(255 * cam)
+        cam = cv2.resize(cam, (224, 224))
+        
         # ✅ Create overlay
         original_img = np.asarray(img.resize((224, 224)))
-        heatmap = cv2.applyColorMap(np.uint8(255 * cam), cv2.COLORMAP_JET)
+        heatmap = cv2.applyColorMap(cam, cv2.COLORMAP_JET)
+        
+        # ✅ Final overlay
         overlay = cv2.addWeighted(original_img, 0.6, heatmap, 0.4, 0)
 
         gradcam_filename = f"gradcam_{timestamp}.png"