Update app.py

app.py

@@ -8,7 +8,7 @@ from huggingface_hub import hf_hub_download
 #     return f"./{model_id}"
 
 
-def yolov9_inference(img_path, model_id, image_size, conf_threshold, iou_threshold):
+def yolov9_inference(img_path, image_size, conf_threshold, iou_threshold):
     """
     Load a YOLOv9 model, configure it, perform inference on an image, and optionally adjust 
     the input size and apply test time augmentation.
@@ -25,7 +25,7 @@ def yolov9_inference(img_path, model_id, image_size, conf_threshold, iou_thresho
     
     # Load the model
     # model_path = download_models()
-    model = yolov9.load(best.pt, device="cuda:0")
+    model = yolov9.load("./best.pt")
     
     # Set model parameters
     model.conf = conf_threshold
@@ -45,7 +45,6 @@ def app():
         with gr.Row():
             with gr.Column():
                 img_path = gr.Image(type="filepath", label="Image")
-                model_path = "best.pt"
                 image_size = gr.Slider(
                     label="Image Size",
                     minimum=320,
@@ -76,7 +75,7 @@ def app():
             fn=yolov9_inference,
             inputs=[
                 img_path,
-                model_path,
+                # model_path,
                 image_size,
                 conf_threshold,
                 iou_threshold,