fcakyon commited on
Commit
e357a0f
1 Parent(s): 61fbf22

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -108,16 +108,16 @@ inputs = [
108
  value="NMS",
109
  label="postprocess_type",
110
  ),
111
- gr.inputs.Dropdown(
112
  ["IOU", "IOS"], type="value", default="IOU", label="postprocess_type"
113
  ),
114
- gr.inputs.Number(default=0.5, label="postprocess_match_threshold"),
115
- gr.inputs.Checkbox(default=True, label="postprocess_class_agnostic"),
116
  ]
117
 
118
  outputs = [
119
- gr.outputs.Image(type="pil", label="YOLOv5s"),
120
- gr.outputs.Image(type="pil", label="YOLOv5s + SAHI"),
121
  ]
122
 
123
  title = "Small Object Detection with SAHI + YOLOv5"
108
  value="NMS",
109
  label="postprocess_type",
110
  ),
111
+ gr.Dropdown(
112
  ["IOU", "IOS"], type="value", default="IOU", label="postprocess_type"
113
  ),
114
+ gr.Number(default=0.5, label="postprocess_match_threshold"),
115
+ gr.Checkbox(default=True, label="postprocess_class_agnostic"),
116
  ]
117
 
118
  outputs = [
119
+ gr.Image(type="pil", label="YOLOv5s"),
120
+ gr.Image(type="pil", label="YOLOv5s + SAHI"),
121
  ]
122
 
123
  title = "Small Object Detection with SAHI + YOLOv5"