feat: 🎉 enhance predict_image function with additional parameters and update example usage

app.py

@@ -2,22 +2,21 @@
 
 import gradio as gr
 import PIL.Image as Image
-
 from ultralytics import ASSETS, YOLO
 
 model = None
 
 
-def predict_image(img, conf_threshold, iou_threshold, model_name):
+def predict_image(img, conf_threshold, iou_threshold, model_name, show_labels, show_conf, imgsz):
     """Predicts objects in an image using a YOLOv8 model with adjustable confidence and IOU thresholds."""
     model = YOLO(model_name)
     results = model.predict(
         source=img,
         conf=conf_threshold,
         iou=iou_threshold,
-        show_labels=True,
-        show_conf=True,
-        imgsz=640,
+        show_labels=show_labels,
+        show_conf=show_conf,
+        imgsz=imgsz,
     )
 
     for r in results:
@@ -33,14 +32,19 @@ iface = gr.Interface(
         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=["yolov8n", "yolov8s", "yolov8m"], label="Model Name", value="yolov8n"),
+        gr.Radio(choices=["yolov8n", "yolov8s", "yolov8m", "yolov8n-seg", "yolov8s-seg", "yolov8m-seg", "yolov8n-pose", "yolov8s-pose", "yolov8m-pose"], label="Model Name", value="yolov8n"),
+        gr.Checkbox(value=True, label="Show Labels"),
+        gr.Checkbox(value=True, label="Show Confidence"),
+        gr.Radio(choices=[320, 640, 1000], label="Image Size", value=640),
     ],
     outputs=gr.Image(type="pil", label="Result"),
     title="Ultralytics Gradio Application 🚀",
     description="Upload images for inference. The Ultralytics YOLOv8n model is used by default.",
     examples=[
-        [ASSETS / "bus.jpg", 0.25, 0.45, "yolov8n.pt"],
-        [ASSETS / "zidane.jpg", 0.25, 0.45, "yolov8n.pt"],
+        [ASSETS / "bus.jpg", 0.25, 0.45, "yolov8n", True, True, 640],
+        [ASSETS / "zidane.jpg", 0.25, 0.45, "yolov8n", True, True, 640],
+        [ASSETS / "bus.jpg", 0.25, 0.45, "yolov8n-seg", True, True, 640],
+        [ASSETS / "zidane.jpg", 0.25, 0.45, "yolov8n-seg", True, True, 640],
     ],
 )
 iface.launch(share=True)

requirements.txt

@@ -1,6 +1,2 @@
 gradio
-torch>=1.8.0
-torch>=1.8.0,!=2.4.0; sys_platform == 'win32'  # Windows CPU errors w/ 2.4.0 https://github.com/ultralytics/ultralytics/issues/15049
-torchvision>=0.9.0
 ultralytics
-Pillow