Update README.md

README.md

@@ -5,16 +5,28 @@ base_model: Qwen/Qwen2-VL-7B-Instruct
 pipeline_tag: image-text-to-text
 ---
 
-This repository contains the model of the paper [OS-ATLAS: A Foundation Action Model for Generalist GUI Agents](https://huggingface.co/papers/2410.23218).
+# OS-Atlas: A Foundation Action Model For Generalist GUI Agents
 
 <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>
 
+## Overview
 ![os-atlas](https://github.com/user-attachments/assets/cf2ee020-5e15-4087-9a7e-75cc43662494)
 
+OS-Atlas provides a series of models specifically designed for GUI agents.
+
+For GUI grounding tasks, you can use:
+- [OS-Atlas-Base-7B](https://huggingface.co/OS-Copilot/OS-Atlas-Base-7B)
+- [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-Pro-7B](https://huggingface.co/OS-Copilot/OS-Atlas-Pro-7B)
+- [OS-Atlas-Pro-4B](https://huggingface.co/OS-Copilot/OS-Atlas-Pro-4B)
+
+
 ## Quick Start
 OS-Atlas-Base-7B is a GUI grounding model finetuned from [Qwen2-VL-7B-Instruct](https://huggingface.co/Qwen/Qwen2-VL-7B-Instruct).
 
@@ -26,6 +38,7 @@ First, ensure that the necessary dependencies are installed:
 pip install transformers
 pip install qwen-vl-utils
 ```
+Then download the [example image](https://github.com/OS-Copilot/OS-Atlas/blob/main/examples/images/web_6f93090a-81f6-489e-bb35-1a2838b18c01.png) and save it to the current directory.
 
 Inference code example:
 ```python
@@ -44,7 +57,7 @@ messages = [
         "content": [
             {
                 "type": "image",
-                "image": "https://github.com/OS-Copilot/OS-Atlas/blob/main/exmaples/images/web_6f93090a-81f6-489e-bb35-1a2838b18c01.png",
+                "image": "./web_6f93090a-81f6-489e-bb35-1a2838b18c01.png",
             },
             {"type": "text", "text": "In this UI screenshot, what is the position of the element corresponding to the command \"switch language of current page\" (with bbox)?"},
         ],