Lasion commited on
Commit
343e9ef
·
1 Parent(s): 4829192
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,6 +9,6 @@ model = YOLO("yolov8n-nckh2023.pt") # Select YOLO model
9
  iface = gr.Interface(
10
  fn=run,
11
  inputs=gr.inputs.Image(label="Upload a image or video", type="filepath"),
12
- outputs=gr.outputs.Image(label="Image output"),
13
  title="Motorcyclist, helmet, and license plate detection")
14
  iface.launch()
 
9
  iface = gr.Interface(
10
  fn=run,
11
  inputs=gr.inputs.Image(label="Upload a image or video", type="filepath"),
12
+ outputs=gr.Image(shape=[256,256], elem_id="output_image").style(width=256, height=256),
13
  title="Motorcyclist, helmet, and license plate detection")
14
  iface.launch()