Update app.py

app.py

@@ -3,7 +3,7 @@ import json
 from datetime import datetime
 import gradio as gr
 import torch
-import spaces
+# import spaces
 from PIL import Image, ImageDraw
 from qwen_vl_utils import process_vision_info
 from transformers import Qwen2VLForConditionalGeneration, AutoProcessor
@@ -83,7 +83,7 @@ def crop_image(image_path, click_xy, crop_factor=0.5):
 
     return cropped_image_path
 
-@spaces.GPU
+# @spaces.GPU
 def run_showui(image, query, session_id, iterations=1):
     """Main function for iterative inference."""
     image_path = array_to_image_path(image, session_id)
@@ -228,11 +228,53 @@ examples = [
     # ["./examples/safari_google.png", "Click on search bar.", True],
 ]
 
+
+
+title_markdown = ("""
+# UGround: Universal Visual Grounding for GUI Agents
+[[🏠Project Homepage](https://osu-nlp-group.github.io/UGround/)] [[Code](https://github.com/OSU-NLP-Group/UGround)] [[😊Model](https://huggingface.co/collections/osunlp/uground-677824fc5823d21267bc9812)][[📚Paper](https://arxiv.org/abs/2410.05243)]
+""")
+
+tos_markdown = ("""
+### Terms of use
+By using this service, users are required to agree to the following terms:
+The service is a research preview intended for non-commercial use only. It only provides limited safety measures and may generate offensive content. It must not be used for any illegal, harmful, violent, racist, or sexual purposes. The service may collect user dialogue data for future research.
+Please click the "Flag" button if you get any inappropriate answer! We will collect those to keep improving our moderator.
+For an optimal experience, please use desktop computers for this demo, as mobile devices may compromise its quality.
+""")
+
+
+learn_more_markdown = ("""
+### License
+The service is a research preview intended for non-commercial use only, subject to the model [License](https://github.com/facebookresearch/llama/blob/main/MODEL_CARD.md) of LLaMA, [Terms of Use](https://openai.com/policies/terms-of-use) of the data generated by OpenAI. Please contact us if you find any potential violation.
+""")
+
+block_css = """
+
+#buttons button {
+    min-width: min(120px,100%);
+}
+
+#chatbot img {
+    max-width: 80%;    /* 宽图片根据宽度调整 */
+    max-height: 80vh;  /* 高图片根据视口高度调整 */
+    width: auto;        /* 保持宽度自适应 */
+    height: auto;       /* 保持高度自适应 */
+    object-fit: contain; /* 保持图片宽高比，不失真 */
+}
+
+"""
+
+
 def build_demo(embed_mode, concurrency_count=1):
-    with gr.Blocks(title="UGround Demo", theme=gr.themes.Default()) as demo:
+    with gr.Blocks(title="UGround Demo", theme=gr.themes.Default(), css=block_css) as demo:
         state_image_path = gr.State(value=None)
         state_session_id = gr.State(value=None)
 
+
+        if not embed_mode:
+            gr.Markdown(title_markdown)
+
         # if not embed_mode:
         #     gr.HTML(
         #         f"""
@@ -272,7 +314,7 @@ def build_demo(embed_mode, concurrency_count=1):
 
                 textbox = gr.Textbox(
                     show_label=True,
-                    placeholder="Enter a query (e.g., 'Click Nahant')",
+                    placeholder="Enter an element description (referring expression) or a single-step instruction and press ENTER",
                     label="Query",
                 )
                 submit_btn = gr.Button(value="Submit", variant="primary")
@@ -365,6 +407,10 @@ def build_demo(embed_mode, concurrency_count=1):
                 queue=False
             )
 
+        if not embed_mode:
+            gr.Markdown(tos_markdown)
+            gr.Markdown(learn_more_markdown)
+
     return demo
 
 if __name__ == "__main__":