aijack commited on
Commit
60abbab
·
1 Parent(s): 0302f6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -48,8 +48,8 @@ def yolov8_inference(
48
 
49
  inputs = [
50
  gr.inputs.Image(type="filepath", label="Input Image"),
51
- gr.inputs.Dropdown(["aijack/yv8/yv8n", "aijack/yv8/yv8m","aijack/yv8x"],
52
- default="aijack/yv8m", label="Model"),
53
  gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
54
  gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.25, step=0.05, label="Confidence Threshold"),
55
  gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold"),
@@ -58,7 +58,7 @@ inputs = [
58
  outputs = gr.outputs.Image(type="filepath", label="Output Image")
59
  title = "Ultralytics YOLOv8: State-of-the-Art YOLO Models"
60
 
61
- examples = [['test1.jpg', 'aijack/yv8m', 640, 0.25, 0.45], ['test2', 'aijack/yv8x', 1280, 0.25, 0.45]]
62
  demo_app = gr.Interface(
63
  fn=yolov8_inference,
64
  inputs=inputs,
 
48
 
49
  inputs = [
50
  gr.inputs.Image(type="filepath", label="Input Image"),
51
+ gr.inputs.Dropdown(["aijack/yv8/yv8n", "aijack/yv8/yv8m","aijack/yv8/yv8x"],
52
+ default="aijack/yv8/yv8m", label="Model"),
53
  gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
54
  gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.25, step=0.05, label="Confidence Threshold"),
55
  gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold"),
 
58
  outputs = gr.outputs.Image(type="filepath", label="Output Image")
59
  title = "Ultralytics YOLOv8: State-of-the-Art YOLO Models"
60
 
61
+ examples = [['test1.jpg', 'aijack/yv8/yv8m', 640, 0.25, 0.45], ['test2', 'aijack/yv8/yv8x', 1280, 0.25, 0.45]]
62
  demo_app = gr.Interface(
63
  fn=yolov8_inference,
64
  inputs=inputs,