Update app.py

app.py

@@ -97,7 +97,7 @@ def run_example(image, text_input, model_id="OS-Copilot/OS-Atlas-Base-7B"):
     boxes = [[boxes[0][0], boxes[0][1], boxes[1][0], boxes[1][1]]]
 
     scaled_boxes = rescale_bounding_boxes(boxes, image.width, image.height)
-    return object_ref, draw_bounding_boxes(image, scaled_boxes)
+    return object_ref, scaled_boxes, draw_bounding_boxes(image, scaled_boxes)
 
 css = """
   #output {
@@ -119,6 +119,7 @@ with gr.Blocks(css=css) as demo:
             submit_btn = gr.Button(value="Submit")
         with gr.Column():
             model_output_text = gr.Textbox(label="Model Output Text")
+            model_output_box = gr.Textbox(label="Model Output Box")
             annotated_image = gr.Image(label="Annotated Image")
 
     gr.Examples(
@@ -127,12 +128,12 @@ with gr.Blocks(css=css) as demo:
             ["assets/web_6f93090a-81f6-489e-bb35-1a2838b18c01.png", "switch to discussions"],
         ],
         inputs=[input_img, text_input],
-        outputs=[model_output_text, annotated_image],
+        outputs=[model_output_text, model_output_box, annotated_image],
         fn=run_example,
         cache_examples=True,
         label="Try examples"
     )
 
-    submit_btn.click(run_example, [input_img, text_input, model_selector], [model_output_text, annotated_image])
+    submit_btn.click(run_example, [input_img, text_input, model_selector], [model_output_text, model_output_box, annotated_image])
 
 demo.launch(debug=True)