skyBluezz commited on
Commit
d678dd8
·
verified ·
1 Parent(s): 3444920

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -335,7 +335,9 @@ def create_demo(model):
335
  with gr.Column():
336
  design_image = gr.Image(label="Output Mask", elem_id='img-display-output')
337
  with gr.Column():
338
- bboxes = gr.JSON(label="Detected Objects", elem_id='json-display-output')
 
 
339
 
340
  def on_submit(image, text, num_steps, guidance_scale, seed, strength, detr_confidence_threshold, a_prompt, n_prompt, img_size):
341
  model.seed = seed
 
335
  with gr.Column():
336
  design_image = gr.Image(label="Output Mask", elem_id='img-display-output')
337
  with gr.Column():
338
+ bboxes = gr.JSON(label="Detected Bboxes", elem_id='json-display-output')
339
+ with gr.Column():
340
+ labels = gr.JSON(label="Detected Objects Labels", elem_id='json-display-output')
341
 
342
  def on_submit(image, text, num_steps, guidance_scale, seed, strength, detr_confidence_threshold, a_prompt, n_prompt, img_size):
343
  model.seed = seed