Debug: yolov8 target lyr

app.py

@@ -4,7 +4,7 @@ from PIL import Image
 import torchvision.transforms as transforms
 import gradio as gr
 from yolov5 import xai_yolov5
-from yolov8 import xai_yolov8
+from yolov8 import xai_yolov8n
 
 def process_image(image, yolo_versions=["yolov5"]):
     image = np.array(image)
@@ -15,7 +15,7 @@ def process_image(image, yolo_versions=["yolov5"]):
         if yolo_version == "yolov5":
             result_images.append(xai_yolov5(image))
         elif yolo_version == "yolov8":
-            result_images.append(xai_yolov8(image))
+            result_images.append(xai_yolov8n(image))
         else:
             result_images.append((Image.fromarray(image), f"{yolo_version} not yet implemented."))
     return result_images