Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import PIL.Image as Image
 
-from ultralytics import ASSETS, YOLO
+from ultralytics import YOLO
 
 model = None
 
@@ -25,20 +25,18 @@ def predict_image(img, conf_threshold, iou_threshold, model_name):
     return im
 
 
-iface = gr.Interface(
+demo = gr.Interface(
     fn=predict_image,
     inputs=[
         gr.Image(type="pil", label="Upload Image"),
-        gr.Slider(minimum=0, maximum=1, value=0.25, label="Confidence threshold"),
-        gr.Slider(minimum=0, maximum=1, value=0.45, label="IoU threshold"),
-        gr.Radio(choices=["best", "yolo11s", "yolo11n-seg", "yolo11s-seg", "yolo11n-pose", "yolo11s-pose"], label="Model Name", value="yolo11n"),
+        # gr.Slider(minimum=0, maximum=1, value=0.25, label="Confidence threshold"),
+        0.24,
+        0.45,
+        # gr.Slider(minimum=0, maximum=1, value=0.45, label="IoU threshold"),
+        gr.Radio(choices=["best", "yolo11s", "yolo11n"], label="Model Name", value="best"),
     ],
     outputs=gr.Image(type="pil", label="Result"),
     title="Ultralytics Gradio Application 🚀",
     description="Upload images for inference. The Ultralytics YOLO11n model is used by default.",
-    examples=[
-        [ASSETS / "bus.jpg", 0.25, 0.45, "yolo11n.pt"],
-        [ASSETS / "zidane.jpg", 0.25, 0.45, "yolo11n.pt"],
-    ],
 )
-iface.launch(share=True)
+demo.launch(share=True)