Update README.md

README.md

@@ -75,11 +75,18 @@ model = AutoModelForCausalLM.from_pretrained(
 )
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
-prompt = "Find the value of $x$ that satisfies the equation $4x+5 = 6x+7$."
+# CoT
 messages = [
-    {"role": "system", "content": "You are Qwen, created by Alibaba Cloud. You are a helpful assistant."},
+    {"role": "system", "content": "Please reason step by step, and put your final answer within \\boxed{}."},
     {"role": "user", "content": prompt}
 ]
+
+# PoT
+messages = [
+    {"role": "system", "content": "Please integrate natural language reasoning with programs to solve the problem above, and put your final answer within \\boxed{}."},
+    {"role": "user", "content": prompt}
+]
+
 text = tokenizer.apply_chat_template(
     messages,
     tokenize=False,