Update app.py

app.py

@@ -137,7 +137,7 @@ def predict():
         input_tensor = transform(img).unsqueeze(0)
 
         # Grad-CAM setup
-        target_layer_name = "features.norm5"
+        target_layer_name = "features.2.local_spatial_conv3"
         gradcam = GradCAM(model, target_layer_name)
 
         # Forward pass