Update app.py

app.py

@@ -3,6 +3,8 @@ import sahi
 import torch
 from ultralyticsplus import YOLO, render_model_output
 
+
+
 model_names = [
     "yolov8n-seg.pt",
     "yolov8s-seg.pt",
@@ -14,12 +16,13 @@ model_names = [
 current_model_name = "yolov8m-seg.pt"
 model = YOLO(current_model_name)
 
+
 def yolov8_inference(
     image: gr.inputs.Image = None,
-    model_name: str = current_model_name,
-    image_size: int = 640,
-    conf_threshold: float = 0.25,
-    iou_threshold: float = 0.45,
+    model_name: gr.inputs.Dropdown = None,
+    image_size: gr.inputs.Slider = 640,
+    conf_threshold: gr.inputs.Slider = 0.25,
+    iou_threshold: gr.inputs.Slider = 0.45,
 ):
     """
     YOLOv8 inference function
@@ -33,8 +36,10 @@ def yolov8_inference(
         Rendered image
     """
     global model
+    global current_model_name
     if model_name != current_model_name:
         model = YOLO(model_name)
+        current_model_name = model_name
     model.overrides["conf"] = conf_threshold
     model.overrides["iou"] = iou_threshold
     results = model.predict(image, imgsz=image_size, return_outputs=True)
@@ -47,8 +52,9 @@ def yolov8_inference(
 
     return renders[0]
 
+
 inputs = [
-    gr.Image(type="file", label="Input Image"),
+    gr.Image(type="filepath", label="Input Image"),
     gr.Dropdown(
         model_names,
         value=current_model_name,
@@ -61,21 +67,21 @@ inputs = [
     gr.Slider(minimum=0.0, maximum=1.0, value=0.45, step=0.05, label="IOU Threshold"),
 ]
 
-outputs = gr.Image(type="file", label="Output Image")
+outputs = gr.Image(type="filepath", label="Output Image")
 title = "Ultralytics YOLOv8 Segmentation Demo"
 
-# 设置默认输入参数
-default_input = ["ikun.jpg", current_model_name, 640, 0.6, 0.45]
+examples = [
+    ["ikun.jpg", "yolov8m-seg.pt", 640, 0.6, 0.45],
+    ["people.jpg", "yolov8m-seg.pt", 640, 0.25, 0.45],
 
+]
 demo_app = gr.Interface(
     fn=yolov8_inference,
     inputs=inputs,
     outputs=outputs,
     title=title,
-    examples=[default_input],
+    examples=examples,
     cache_examples=True,
     theme="default",
 )
-#
-# 运行应用，并设置live=True
-demo_app.launch(debug=True, enable_queue=True, live=True)
+demo_app.launch(debug=True, enable_queue=True)