Update app.py

app.py

@@ -109,31 +109,30 @@ css = """
 with gr.Blocks(css=css) as demo:
     gr.Markdown(
     """
-    # OS-Atlas Demo
+    # Demo for OS-ATLAS: A Foundation Action Model For Generalist GUI Agents
     """)
-    with gr.Tab(label="OS-Atlas Input"):
-        with gr.Row():
-            with gr.Column():
-                input_img = gr.Image(label="Input Image", type="pil")
-                model_selector = gr.Dropdown(choices=list(models.keys()), label="Model", value="OS-Copilot/OS-Atlas-Base-7B")
-                text_input = gr.Textbox(label="User Prompt")
-                submit_btn = gr.Button(value="Submit")
-            with gr.Column():
-                model_output_text = gr.Textbox(label="Model Output Text")
-                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, 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])
+    with gr.Row():
+        with gr.Column():
+            input_img = gr.Image(label="Input Image", type="pil")
+            model_selector = gr.Dropdown(choices=list(models.keys()), label="Model", value="OS-Copilot/OS-Atlas-Base-7B")
+            text_input = gr.Textbox(label="User Prompt")
+            submit_btn = gr.Button(value="Submit")
+        with gr.Column():
+            model_output_text = gr.Textbox(label="Model Output Text")
+            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, 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])
 
 demo.launch(debug=True)