Update app.py

app.py

@@ -3,12 +3,12 @@ import cv2
 import tempfile
 from ultralytics import YOLOv10
 import supervision as sv
-import spaces
+from huggingface_hub import hf_hub_download
 
 
 def download_models(model_id):
-    model_path = f'kadirnar/{model_id}'
-    return model_path
+    hf_hub_download("jameslahm/yolov10x", filename=f"{model_id}", local_dir=f"./")
+    return f"./{model_id}"
     
 box_annotator = sv.BoxAnnotator()
 category_dict = {
@@ -31,10 +31,9 @@ category_dict = {
 }
 
 
-@spaces.GPU(duration=200)
 def yolov10_inference(image, video, model_id, image_size, conf_threshold, iou_threshold):
     model_path = download_models(model_id)
-    model = YOLOv10.from_pretrained(model_path)
+    model = YOLOv10(model_path)
     
     if image:
         results = model(source=image, imgsz=image_size, iou=iou_threshold, conf=conf_threshold, verbose=False)[0]
@@ -100,14 +99,14 @@ def app():
                 model_id = gr.Dropdown(
                     label="Model",
                     choices=[
-                        "yolov10n",
-                        "yolov10s",
-                        "yolov10m",
-                        "yolov10b",
-                        "yolov10l",
-                        "yolov10x",
+                        "yolov10n.pt",
+                        "yolov10s.pt",
+                        "yolov10m.pt",
+                        "yolov10b.pt",
+                        "yolov10l.pt",
+                        "yolov10x.pt",
                     ],
-                    value="yolov10m",
+                    value="yolov10m.pt",
                 )
                 image_size = gr.Slider(
                     label="Image Size",
@@ -167,14 +166,14 @@ def app():
             examples=[
                 [
                     "ultralytics/assets/bus.jpg",
-                    "yolov10s",
+                    "yolov10s.pt",
                     640,
                     0.25,
                     0.45,
                 ],
                 [
                     "ultralytics/assets/zidane.jpg",
-                    "yolov10s",
+                    "yolov10s.pt",
                     640,
                     0.25,
                     0.45,
@@ -212,4 +211,4 @@ with gradio_app:
             app()
 
 if __name__ == '__main__':
-    gradio_app.launch()
+    gradio_app.launch()