Fix: yolo8 loading issue.

app.py

@@ -10,6 +10,7 @@ from pytorch_grad_cam import EigenCAM
 from pytorch_grad_cam.utils.image import show_cam_on_image, scale_cam_image
 from PIL import Image
 import gradio as gr
+from ultralytics import YOLO
 
 # Global Color Palette
 COLORS = np.random.uniform(0, 255, size=(80, 3))
@@ -45,7 +46,7 @@ def load_yolo_model(version="yolov5"):
     if version == "yolov5":
         model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
     elif version == "yolov8":
-        model = torch.hub.load('ultralytics/yolov8', 'yolov8s', pretrained=True)  # YOLOv8 is part of the yolov5 repo starting from v7.0
+        model = YOLO("yolov8n.pt")  # YOLOv8 is part of the yolov5 repo starting from v7.0
     elif version == "yolov10":
         model = torch.hub.load('ultralytics/yolov5', 'yolov5m', pretrained=True)  # Placeholder for YOLOv10 (use an appropriate version if available)
     else: