Update README.md

README.md

@@ -62,8 +62,8 @@ model = PeftModel.from_pretrained(base_model, "PanzerBread/promptcot-p")
 tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-7B-Instruct")
 
 concepts = "algebra | quadratic equations"
-rationale = "We need to find the roots of a quadratic equation..."
-prompt = f"Concepts: {concepts}\nRationale: {rationale}\nProblem:"
+# It will think about the concepts, and then generate a problem after "Problem: "
+prompt = f"Concepts: {concepts}\nRationale:"
 
 inputs = tokenizer(prompt, return_tensors="pt")
 outputs = model.generate(**inputs, max_new_tokens=256)