Update app.py

app.py

@@ -357,7 +357,7 @@ def create_demo(model):
             
         return out_img, bboxes.tolist(), labels
 
-    submit.click(on_submit, inputs=[input_image, input_text, num_steps, guidance_scale, seed, strength, detr_confidence_threshold, a_prompt, n_prompt, img_size], outputs=[design_image, bboxes])
+    submit.click(on_submit, inputs=[input_image, input_text, num_steps, guidance_scale, seed, strength, detr_confidence_threshold, a_prompt, n_prompt, img_size], outputs=[design_image, bboxes, labels])
     
 
 controlnet_depth= ControlNetModel.from_pretrained(