Update README.md

README.md

@@ -17,11 +17,61 @@ XtraGPT is a series of LLMs for Human-AI Collaboration on Controllable Scientifi
 
 ## Requirements
 
-The code of Qwen2.5 has been in the latest Hugging face `transformers` and we advise you to use the latest version of `transformers`.
+The code of XtraGPT has been in the latest Hugging face `transformers` and we advise you to use the latest version of `transformers`.
 
 ## Quickstart
 
-Here provides a code snippet with `apply_chat_template` to show you how to load the tokenizer and model and how to generate contents.
+```python
+from openai import OpenAI
+model_name = "Xtra-Computing/XtraGPT-7B"
+client = OpenAI(
+    base_url="http://localhost:8088/v1",
+    api_key="sk-1234567890"
+)
+
+paper_content="markdown"
+selected_content="After that, we define CAT-score to measure the matching degree between the filtered attention matrix and the distance matrix."
+
+prompt = "help me redefine cat-score based on the context."
+
+content = f"""
+Please improve the selected content based on the following. Act as an expert model for improving articles **PAPER_CONTENT**.\n
+The output needs to answer the **QUESTION** on **SELECTED_CONTENT** in the input. Avoid adding unnecessary length, unrelated details, overclaims, or vague statements.
+Focus on clear, concise, and evidence-based improvements that align with the overall context of the paper.\n
+<PAPER_CONTENT> {paper_content} </PAPER_CONTENT>\n <SELECTED_CONTENT> {selected_content} </SELECTED_CONTENT>\n <QUESTION> {prompt} </QUESTION>\n
+"""
+
+response = client.chat.completions.create(
+    model="xtragpt",
+    messages=[{"role": "user", "content": content}],
+    temperature=0.7,
+    max_tokens=16384
+)
+print(response.choices[0].message.content)
+
+```
+
+## Citation
+
+If you find our work helpful, feel free to give us a cite.
+
+```
+@misc{xtracomputing2025xtraqa,
+    title = {XtraQA},
+    url = {https://huggingface.co/Xtra-Computing/XtraGPT-7B},
+    author = {Xtra Computing Group},
+    year = {2025}
+}
+@article{xtracomputing2025xtragpt,
+      title={XtraGPT: LLMs for Human-AI Collaboration on Controllable Scientific Paper Refinement}, 
+      author={Xtra Computing Group},
+      journal={arXiv preprint arXiv:abcdefg},
+      year={2025}
+}
+```
+
+
+
 
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
@@ -48,7 +98,6 @@ Finally, the CAT-score is used to interpret how CodePTMs attend code structure,
 selected_content="""
 After that, we define CAT-score to measure the matching degree between the filtered attention matrix and the distance matrix.
 """
-
 prompt ="""
 help me redefine cat-score based on the context.
 """
@@ -92,68 +141,5 @@ response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
 ```
 
 
-```python
-from openai import OpenAI
-model_name = "Xtra-Computing/XtraGPT-7B"
-client = OpenAI(
-    base_url="http://localhost:8088/v1",
-    api_key="sk-1234567890"
-)
-
-paper_content="""
-markdown
-"""
-selected_content="""
-After that, we define CAT-score to measure the matching degree between the filtered attention matrix and the distance matrix.
-"""
-
-prompt ="""
-help me redefine cat-score based on the context.
-"""
-
-content = f"""
-Please improve the selected content based on the following. Act as an expert model for improving articles **PAPER_CONTENT**.\n
-The output needs to answer the **QUESTION** on **SELECTED_CONTENT** in the input. Avoid adding unnecessary length, unrelated details, overclaims, or vague statements.
-Focus on clear, concise, and evidence-based improvements that align with the overall context of the paper.\n
-
-<PAPER_CONTENT>
-{paper_content}
-</PAPER_CONTENT>\n
-
-<SELECTED_CONTENT>
-{selected_content}
-</SELECTED_CONTENT>\n
 
-<QUESTION>
-{prompt}
-</QUESTION>\n
-"""
 
-response = client.chat.completions.create(
-    model="xtragpt",
-    messages=[{"role": "user", "content": content}],
-    temperature=0.7,
-    max_tokens=16384
-)
-print(response.choices[0].message.content)
-
-```
-
-## Citation
-
-If you find our work helpful, feel free to give us a cite.
-
-```
-@misc{xtracomputing2025xtraqa,
-    title = {XtraQA},
-    url = {https://huggingface.co/Xtra-Computing/XtraGPT-7B},
-    author = {Xtra Computing Group},
-    year = {2025}
-}
-@article{xtracomputing2025xtragpt,
-      title={XtraGPT: LLMs for Human-AI Collaboration on Controllable Scientific Paper Refinement}, 
-      author={Xtra Computing Group},
-      journal={arXiv preprint arXiv:abcdefg},
-      year={2025}
-}
-```