Debug: refactor src

yolov8.py

@@ -55,13 +55,12 @@ def xai_yolov8s(image):
     model.eval()
     results = model(image)
     detections = results[0]
-    
     boxes, colors, names = parse_detections(detections, model)
     detections_img = draw_detections(boxes, colors, names, image.copy())
     img_float = np.float32(image) / 255
     transform = transforms.ToTensor()
     tensor = transform(img_float).unsqueeze(0)
-    target_layers = [model.model[-2]]  # Adjust according to YOLOv8 architecture
+    target_layers = [model.model.model[-2]]  # Adjust to YOLOv8 architecture
     cam_image, renormalized_cam_image = generate_cam_image(model.model, target_layers, tensor, image, boxes)
     final_image = np.hstack((image, cam_image, renormalized_cam_image))
     caption = "Results using YOLOv8"