Fix:model access.

app.py

@@ -75,7 +75,7 @@ def process_image(image, yolo_versions=["yolov5"]):
     for yolo_version in yolo_versions:
         # Load the model based on YOLO version
         model = load_yolo_model(yolo_version)
-        target_layers = [model.model.model.model[-2]]  # Assumes last layer is used for Grad-CAM
+        target_layers = [model.model[-1]]  # Assumes last layer is used for Grad-CAM
 
         # Run YOLO detection
         results = model([rgb_img])