Update app.py

app.py

@@ -90,8 +90,6 @@ def run_example(image, text_input, model_id="OS-Copilot/OS-Atlas-Base-7B"):
     object_ref_pattern = r"<\|object_ref_start\|>(.*?)<\|object_ref_end\|>"
     box_pattern = r"<\|box_start\|>(.*?)<\|box_end\|>"
 
-    print(text)
-    print(re.search(object_ref_pattern, text))
     object_ref = re.search(object_ref_pattern, text).group(1)
     box_content = re.search(box_pattern, text).group(1)
 
@@ -99,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 output_text, boxes, draw_bounding_boxes(image, scaled_boxes)
+    return object_ref, draw_bounding_boxes(image, scaled_boxes)
 
 css = """
   #output {
@@ -122,20 +120,20 @@ 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")
-                parsed_boxes = gr.Textbox(label="Parsed Boxes")
                 annotated_image = gr.Image(label="Annotated Image")
 
         gr.Examples(
             examples=[
                 ["assets/web_6f93090a-81f6-489e-bb35-1a2838b18c01.png", "select search textfield"],
+                ["assets/web_6f93090a-81f6-489e-bb35-1a2838b18c01.png", "switch to discussions"],
             ],
             inputs=[input_img, text_input],
-            outputs=[model_output_text, parsed_boxes, annotated_image],
+            outputs=[model_output_text, 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, parsed_boxes, annotated_image])
+        submit_btn.click(run_example, [input_img, text_input, model_selector], [model_output_text, annotated_image])
 
 demo.launch(debug=True)