Debug: yolov8 target lyr

yolov8.py

@@ -48,7 +48,7 @@ def xai_yolov8(image):
     image_tensor = torch.tensor(np.array(image)).permute(2, 0, 1).unsqueeze(0).float() / 255.0
     image_tensor = image_tensor.to('cpu')
     print(model.model)  # Output model layers to find the target layer
-    grad_cam = GradCAM(model.model, target_layer='model.model.model[-4]')  
+    grad_cam = GradCAM(model.model, target_layer='model.model[-1]')  
 
     # Perform Grad-CAM
     cam_map = grad_cam(image_tensor)