Update app.py

app.py

@@ -122,7 +122,7 @@ def predict():
         confidence = float(probabilities[class_index])
 
         # ✅ Grad-CAM++ setup
-        target_layer = model.conv1x1  # Final block of EfficientNet feature extractor
+        target_layer = model.swin_attn # Final block of EfficientNet feature extractor
         cam_model = GradCAMPlusPlus(model=model, target_layers=[target_layer])
 
         cam_output = cam_model(input_tensor=input_tensor,