Update README.md

README.md

@@ -89,7 +89,7 @@ pipeline = transformers.pipeline(
 
 </details>
 
-### Text Generation
+### 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|:---|:----------|:--------|:-----:|."
@@ -105,6 +105,7 @@ outputs = pipeline(prompt, max_new_tokens=1000, do_sample=True, temperature=0.7,
 print("***Generation:\n", outputs[0]["generated_text"][len(prompt):])
 ```
 
+```
 ***Prompt:
 <|im_start|>system
 You are a helpful assistant.<|im_end|>
@@ -114,14 +115,15 @@ Respond as succinctly as possible. Format the response as a completion of this t
 |step|subquestion|procedure|result|
 |:---|:----------|:--------|:-----:|.<|im_end|>
 <|im_start|>assistant
+```
 
-***Generation:
+### Output
 
-|step|subquestion|procedure|result|
-|:---|:----------|:--------|:-----:|
+```
+***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|
-
+```