Improve metadata (pipeline tag, library name, correct typo) and add GitHub link

README.md

@@ -1,21 +1,22 @@
 ---
-license: cc-by-nc-sa-4.0
+base_model:
+- Qwen/Qwen2.5-14B-Instruct
 language:
 - en
 - zh
-base_model:
-- Qwen/Qwen2.5-14B-Instruct
+license: cc-by-nc-sa-4.0
+pipeline_tag: translation
+library_name: transformers
 tags:
-- machine tranlsation
+- machine translation
 - O1-like model
 - Chat
-pipeline_tag: text-generation
 ---
 
 # DRT
 
 <p align="center">
-🤗 <a href="https://huggingface.co/Krystalan/DRT-7B">DRT-7B</a>&nbsp&nbsp | &nbsp&nbsp🤗 <a href="https://huggingface.co/Krystalan/DRT-8B">DRT-8B</a>&nbsp&nbsp | &nbsp&nbsp🤗 <a href="https://huggingface.co/Krystalan/DRT-14B">DRT-14B</a>&nbsp&nbsp | &nbsp&nbsp 📑 <a href="https://arxiv.org/abs/2412.17498">Paper</a>
+🤗 <a href="https://huggingface.co/Krystalan/DRT-7B">DRT-7B</a>&nbsp&nbsp | &nbsp&nbsp🤗 <a href="https://huggingface.co/Krystalan/DRT-8B">DRT-8B</a>&nbsp&nbsp | &nbsp&nbsp🤗 <a href="https://huggingface.co/Krystalan/DRT-14B">DRT-14B</a>&nbsp&nbsp | &nbsp&nbsp 📑 <a href="https://arxiv.org/abs/2412.17498">Paper</a> | &nbsp;&nbsp 💻 <a href="https://github.com/krystalan/DRT-o1">Code</a>
 
 </p>
 
@@ -80,7 +81,8 @@ In this work, we introduce DRT, an attempt to bring the success of long thought
 ### Model Prompts
 During model inference, please use the following prompts:
 - System prompt: `You are a philosopher skilled in deep thinking, accustomed to exploring complex problems with profound insight.`
-- User prompt: `Please translate the following text from English to Chinese:\n[An English text]`
+- User prompt: `Please translate the following text from English to Chinese:
+[An English text]`
 
 DRT models will first generate the thought and then provide the final translation, with the following format:
 ```
@@ -107,7 +109,8 @@ model = AutoModelForCausalLM.from_pretrained(
 )
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
-prompt = "Please translate the following text from English to Chinese:\nThe mother, with her feet propped up on a stool, seemed to be trying to get to the bottom of that answer, whose feminine profundity had struck her all of a heap."
+prompt = "Please translate the following text from English to Chinese:
+The mother, with her feet propped up on a stool, seemed to be trying to get to the bottom of that answer, whose feminine profundity had struck her all of a heap."
 messages = [
     {"role": "system", "content": "You are a philosopher skilled in deep thinking, accustomed to exploring complex problems with profound insight."},
     {"role": "user", "content": prompt}
@@ -154,7 +157,8 @@ chat_response = client.chat.completions.create(
     model=[model_name],
     messages=[
         {"role": "system", "content": "You are a philosopher skilled in deep thinking, accustomed to exploring complex problems with profound insight."},
-        {"role": "user", "content": "Please translate the following text from English to Chinese:\nThe mother, with her feet propped up on a stool, seemed to be trying to get to the bottom of that answer, whose feminine profundity had struck her all of a heap."},
+        {"role": "user", "content": "Please translate the following text from English to Chinese:
+The mother, with her feet propped up on a stool, seemed to be trying to get to the bottom of that answer, whose feminine profundity had struck her all of a heap."},
     ],
     temperature=0.1,
     top_p=0.8,
@@ -176,10 +180,10 @@ print("Chat response:", chat_response)
 |This cold officer upon a monument, who dropped epithets unconcernedly down, would be finer as a dead man, he thought. | 他认为，这个站在纪念碑上的冷漠官员，若死了会更好，他不带任何感情地抛下了一些称呼。 | 这个冷冰冰的官员站在纪念碑上，毫不在意地抛下一些称号，他想，如果作为一个死人会更出色。 | 纪念碑上的冷淡官员，漫不经心地吟咏那些修饰语，他心想，若化为亡者，或许更显尊贵。 |
 
 
+## Data
 
-## License
-This work is licensed under cc-by-nc-sa-4.0
-
-
+We release the synthesized data (named ```MetaphorTrans```), please refer to `data/MetaphorTrans_*.jsonl`, where `text` and `trans` denote the source English sentences and the target Chinese translations, respectively. `thought` indicates the thought content for MT.
 
 
+# License
+This work is licensed under cc-by-nc-sa-4.0