Debug: refactor src

yolov8.py

@@ -6,6 +6,7 @@ import torchvision.transforms as transforms
 from pytorch_grad_cam import EigenCAM
 from pytorch_grad_cam.utils.image import show_cam_on_image, scale_cam_image
 import gradio as gr
+from ultralytics import YOLO
 
 # Global Color Palette
 COLORS = np.random.uniform(0, 255, size=(80, 3))
@@ -53,7 +54,7 @@ def generate_cam_image(model, target_layers, tensor, rgb_img, boxes):
 
 def xai_yolov8s(image):
     # Load YOLOv8 model
-    model = torch.hub.load('ultralytics/yolov8', 'yolov8s', pretrained=True)
+    model = YOLO('yolov8s.pt')
     model.eval()
     model.cpu()