Fix: resolved yolo model dependency

app.py

@@ -44,14 +44,10 @@ def draw_detections(boxes, colors, names, img):
 
 # Load the appropriate YOLO model based on the version
 def load_yolo_model(version="yolov5"):
-    if version == "yolov3":
-        model = torch.hub.load('ultralytics/yolov3', 'yolov3', pretrained=True)
-    elif version == "yolov5":
+    if version == "yolov5":
         model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
-    elif version == "yolov7":
-        model = torch.hub.load('WongKinYiu/yolov7', 'yolov7', pretrained=True)
     elif version == "yolov8":
-        model = torch.hub.load('ultralytics/yolov5:v7.0', 'yolov5', 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:
@@ -111,7 +107,7 @@ interface = gr.Interface(
     inputs=[
         gr.Image(type="pil", label="Upload an Image"),
         gr.CheckboxGroup(
-            choices=["yolov3", "yolov5", "yolov7", "yolov8", "yolov10"],
+            choices=["yolov5",  "yolov8", "yolov10"],
             value=["yolov5"],  # Set the default value (YOLOv5 checked by default)
             label="Select Model(s)",
         )