Update README.md

README.md

@@ -92,37 +92,33 @@ pipeline = transformers.pipeline(
 ### Run
 
 ```python
-question = "The bakers at the Beverly Hills Bakery baked 200 loaves of bread on Monday morning. They sold 93 loaves in the morning and 39 loaves in the afternoon. A grocery store then returned 6 unsold loaves back to the bakery. How many loaves of bread did the bakery have left?\nRespond as succinctly as possible. Format the response as a completion of this table.\n|step|subquestion|procedure|result|\n|:---|:----------|:--------|:-----:|."
+question = """The bakers at the Beverly Hills Bakery baked 200 loaves of bread on Monday morning. 
+They sold 93 loaves in the morning and 39 loaves in the afternoon. 
+A grocery store then returned 6 unsold loaves back to the bakery. 
+How many loaves of bread did the bakery have left?
+Respond as succinctly as possible. Format the response as a completion of this table:
+|step|subquestion|procedure|result|
+|:---|:----------|:--------|:-----:|"""
+
 
 messages = [
     {"role": "system", "content": "You are a helpful assistant."},
     {"role": "user", "content": question},
 ]
 prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
-print("***Prompt:\n", prompt)
+# print("***Prompt:\n", prompt)
 
 outputs = pipeline(prompt, max_new_tokens=1000, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
-print("***Generation:\n", outputs[0]["generated_text"][len(prompt):])
-```
+print("***Generation:")
+print(outputs[0]["generated_text"][len(prompt):])
 
-### Output
-
-```
-***Prompt:
-<|im_start|>system
-You are a helpful assistant.<|im_end|>
-<|im_start|>user
-The bakers at the Beverly Hills Bakery baked 200 loaves of bread on Monday morning. They sold 93 loaves in the morning and 39 loaves in the afternoon. A grocery store then returned 6 unsold loaves back to the bakery. How many loaves of bread did the bakery have left?
-Respond as succinctly as possible. Format the response as a completion of this table.
-|step|subquestion|procedure|result|
-|:---|:----------|:--------|:-----:|.<|im_end|>
-<|im_start|>assistant
 ```
 
 ```
 ***Generation:
-|1|Calculate total loaves sold|Add morning and afternoon sales|132|
-|2|Calculate remaining loaves after sales|Subtract total sold from total baked|68|
-|3|Calculate final number of loaves|Add returns to remaining loaves|74|
+|1|Initial loaves|Start with total loaves|200|
+|2|Sold in morning|Subtract morning sales|200 - 93 = 107|
+|3|Sold in afternoon|Subtract afternoon sales|107 - 39 = 68|
+|4|Returned loaves|Add returned loaves|68 + 6 = 74|
 ```