Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import spaces
 import supervision as sv
 import PIL.Image as Image
-from ultralytics import YOLO, YOLOv10
+from ultralytics import YOLO
 from huggingface_hub import hf_hub_download, HfApi
 
 global repo_id
@@ -32,10 +32,7 @@ category_dict = {0: 'A', 1: 'B', 2: 'C', 3: 'D', 4: 'E', 5: 'F', 6: 'G', 7: 'H',
 @spaces.GPU
 def yolo_inference(image, model_id, conf_threshold, iou_threshold, max_detection):
     model_path = download_models(model_id)
-    if model_id[:7] == 'yolov10':
-        model = YOLOv10(model_path)
-    else:
-        model = YOLO(model_path)
+    model = YOLO(model_path)
     results = model(source=image, imgsz=640, iou=iou_threshold, conf=conf_threshold, verbose=False, max_det=max_detection)[0]
     detections = sv.Detections.from_ultralytics(results)