linhcuem commited on
Commit
754c31a
1 Parent(s): bdbb930

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -37,20 +37,20 @@ def yolov8_img_inference(
37
 
38
  return render
39
 
40
- inputs_images = [
41
  gr.inputs.Image(type="filepath", label="Input Image"),
42
- gr.inputs.Dropdown(["linhcuem/chamdiem_yolov8_ver10"],
43
- default="linhcuem/chamdiem_yolov8_ver10", label="Model"),
44
  gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
45
- gr.inputs.Slider(minimum=0.0, maximum=1.0, default= 0.25, step=0.05, label="Confidence Threshold"),
46
- gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold"),
47
  ]
48
 
49
- output_image = gr.outputs.Image(type="filepath", label="Output Image")
50
  interface_image = gr.Interface(
51
  fn=yolov8_img_inference,
52
- inputs=inputs_images,
53
- outputs=output_image,
54
  title=model_heading,
55
  description=description,
56
  examples=image_path,
 
37
 
38
  return render
39
 
40
+ inputs_image = [
41
  gr.inputs.Image(type="filepath", label="Input Image"),
42
+ gr.inputs.Dropdown(["linhcuem/chamdiem_yolov8_ver10"],
43
+ default="linhcuem/chamdiem_yolov8_ver10", label="Model"),
44
  gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
45
+ gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.25, step=0.05, label="Confidence Threshold"),
46
+ gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.05, label="IOU Threshold"),
47
  ]
48
 
49
+ outputs_image =gr.outputs.Image(type="filepath", label="Output Image")
50
  interface_image = gr.Interface(
51
  fn=yolov8_img_inference,
52
+ inputs=inputs_image,
53
+ outputs=outputs_image,
54
  title=model_heading,
55
  description=description,
56
  examples=image_path,