Debug: yolov8 target lyr

yolov8.py

@@ -47,11 +47,8 @@ def xai_yolov8(image):
     # Convert image to PyTorch tensor for Grad-CAM
     image_tensor = torch.tensor(np.array(image)).permute(2, 0, 1).unsqueeze(0).float() / 255.0
     image_tensor = image_tensor.to('cpu')
-    
-    # Inspect the model and print layer structure
+    model(image_tensor)  # Forward pass
     print(model.model)  # Output model layers to find the target layer
-    
-    # Find the correct target layer
     target_layer = model.model.model[-1]  # Target the last detection layer
     
     # Initialize Grad-CAM with the target layer