Update app.py

app.py

@@ -95,7 +95,7 @@ class GradCAM:
         weights = self.gradients.mean(dim=(2, 3), keepdim=True)
         cam = (weights * self.activations).sum(dim=1, keepdim=True)
         cam = F.relu(cam)
-        cam = cam.squeeze().cpu().numpy()
+        cam = cam.squeeze().cpu().detach().numpy()
         cam = (cam - cam.min()) / (cam.max() - cam.min() + 1e-8)
         return cam
 
@@ -153,6 +153,8 @@ def predict():
         model.zero_grad()
         output[0, class_index].backward()
         cam = gradcam.generate(class_index)
+        
+
 
         # ✅ Create overlay
         original_img = np.asarray(img.resize((224, 224)))