Update README.md

README.md

@@ -85,12 +85,12 @@ model = PeftModel.from_pretrained(model, "adityasihag/math_QA-gemma-2B-QLoRA-ada
 model = model.merge_and_unload()
 tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
 tokenizer.pad_token = tokenizer.eos_token
-question = """Solve the linear equations. $3(x+2)-x=x + 9$. Find the value of x."""
+question = """Gretchen has 110 coins. There are 30 more gold coins than silver coins. How many gold coins does Gretchen have?"""
 sample_input = f"""Question: {question} \n Answer: """
 sample_input_tokenised = tokenizer(sample_input, return_tensors = "pt").to("cuda")
 generated_ids = model.generate(
                     **sample_input_tokenised,
-                    max_new_tokens = 1024,
+                    max_new_tokens = 512,
                     temperature = 0.3
                 )
 output = tokenizer.decode(generated_ids[0], skip_special_tokens = True)
@@ -99,16 +99,19 @@ print(output)
 
 ##### Sample Input:
 ```
-Question: Solve the linear equations. $3(x+2)-x=x + 9$. Find the value of x. \n Answer: 
+Question: Gretchen has 110 coins. There are 30 more gold coins than silver coins. How many gold coins does Gretchen have? \n Answer: 
 ```
 
 ##### Model Output:
 ```
-Given the linear equation 3(x+2)-x=x+9. 
-First, distribute the 3 in the brackets to get 3x + 6 - x = x + 9. 
-Simplify the equation to get 2x + 6 = x + 9. 
-Next, transpose x from the right side to the left side and from the left side to the right side to get x = 9 - 6. 
-Finally, solve for x to get x = 3.
+Let's assume the number of silver coins is x.
+Since there are 30 more gold coins than silver coins, the number of gold coins is x + 30.
+The total number of coins is x + (x + 30) = 110.
+Combining like terms, we get 2x + 30 = 110.
+Subtracting 30 from both sides, we get 2x = 80.
+Dividing both sides by 2, we get x = 40.
+So, Gretchen has 40 silver coins and 40 + 30 = 70 gold coins.
+The answer is: 70
 ```
 
 #### Prompt Template: