Update README.md

README.md

@@ -9,7 +9,7 @@ pipeline_tag: image-text-to-text
 
 <div align="center">
 
-[\[🏠Homepage\]](https://osatlas.github.io) [\[💻Code\]](https://github.com/OS-Copilot/OS-Atlas) [\[🚀Quick Start\]](#quick-start) [\[📝Paper\]](https://arxiv.org/abs/2410.23218) [\[🤗Models\]](https://huggingface.co/collections/OS-Copilot/os-atlas-67246e44003a1dfcc5d0d045) [\[🤗ScreenSpot-v2\]](https://huggingface.co/datasets/OS-Copilot/ScreenSpot-v2) 
+[\[🏠Homepage\]](https://osatlas.github.io) [\[💻Code\]](https://github.com/OS-Copilot/OS-Atlas) [\[🚀Quick Start\]](#quick-start) [\[📝Paper\]](https://arxiv.org/abs/2410.23218) [\[🤗Models\]](https://huggingface.co/collections/OS-Copilot/os-atlas-67246e44003a1dfcc5d0d045)[\[🤗Data\]](https://huggingface.co/datasets/OS-Copilot/OS-Atlas-data) [\[🤗ScreenSpot-v2\]](https://huggingface.co/datasets/OS-Copilot/ScreenSpot-v2) 
 
 </div>
 
@@ -23,14 +23,16 @@ For GUI grounding tasks, you can use:
 - [OS-Atlas-Base-4B](https://huggingface.co/OS-Copilot/OS-Atlas-Base-4B)
 
 For generating single-step actions in GUI agent tasks, you can use:
-- [OS-Atlas-Action-7B](https://huggingface.co/OS-Copilot/OS-Atlas-Action-7B)
-- [OS-Atlas-Action-4B](https://huggingface.co/OS-Copilot/OS-Atlas-Action-4B)
+- [OS-Atlas-Pro-7B](https://huggingface.co/OS-Copilot/OS-Atlas-Pro-7B)
+- [OS-Atlas-Pro-4B](https://huggingface.co/OS-Copilot/OS-Atlas-Pro-4B)
 
 
 ## OS-Atlas-Action-7B 
 
 `OS-Atlas-Action-7B` is a GUI action model finetuned from OS-Atlas-Base-7B. By taking as input a system prompt, basic and custom actions, and a task instruction, the model generates thoughtful reasoning (`thought`) and executes the appropriate next step (`action`).
 
+Note that the released `OS-Atlas-Pro-7B model` is described in the Section 5.4 of the paper. Compared to the OS-Atlas model in Tables 4 and 5, the Pro model demonstrates superior generalizability and performance. Critically, it is not constrained to specific tasks or training datasets merely to satisfy particular experimental conditions like OOD and SFT. Furthermore, this approach prevents us from overdosing HuggingFace by uploading over 20+ distinct model checkpoints.
+
 ### Installation
 To use `OS-Atlas-Action-7B`, first install the necessary dependencies:
 ```bash
@@ -39,8 +41,9 @@ pip install qwen-vl-utils
 ```
 
 ### Example Inference Code
-Below is an example of how to perform inference using the model:
+First download the [example image](https://github.com/OS-Copilot/OS-Atlas/blob/main/examples/images/action_example_1.jpg) and save it to the current directory.
 
+Below is an example of how to perform inference using the model:
 ```python
 from transformers import Qwen2VLForConditionalGeneration, AutoProcessor
 from qwen_vl_utils import process_vision_info
@@ -134,7 +137,7 @@ messages = [
             },
             {
                 "type": "image",
-                "image": "https://github.com/OS-Copilot/OS-Atlas/blob/main/exmaples/images/action_example_1.jpg",
+                "image": "./action_example_1.jpg",
             },
             {"type": "text", "text": "Task instruction: to allow the user to enter their first name\nHistory: null" },
         ],