kasper-boy commited on
Commit
55e3e8f
1 Parent(s): b11a9e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -57,13 +57,11 @@ def object_detection(image, confidence_threshold):
57
  demo = gr.Interface(
58
  object_detection,
59
  [
60
- gr.inputs.Image(type="pil", label="Upload an Image"),
 
 
61
  gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.5, label="Confidence Threshold")
62
  ],
63
- [
64
- gr.outputs.Image(type="pil", label="Detected Objects"),
65
- gr.outputs.Textbox(label="Detected Objects List")
66
- ],
67
  title="Object Detection with DETR (ResNet-101)",
68
  description="Upload an image and adjust the confidence threshold to view detected objects."
69
  )
 
57
  demo = gr.Interface(
58
  object_detection,
59
  [
60
+ gr.inputs.Image(label="Upload an Image"),
61
+ gr.outputs.Image(label="Detected Objects"),
62
+ gr.outputs.Textbox(label="Detected Objects List"),
63
  gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.5, label="Confidence Threshold")
64
  ],
 
 
 
 
65
  title="Object Detection with DETR (ResNet-101)",
66
  description="Upload an image and adjust the confidence threshold to view detected objects."
67
  )