Update README.md

README.md

@@ -93,7 +93,11 @@ messages = [
 ]
 
 print("\n\n*** Prompt:")
-prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
+prompt = tokenizer.apply_chat_template(
+    messages,
+    tokenize=False,
+    add_generation_prompt=True
+)
 print(prompt)
 
 print("\n\n*** Generate:")
@@ -166,8 +170,9 @@ print(prompt)
 
 <summary>Output</summary>
 
-**Prompt**: <s>[INST] What is your favourite condiment? [/INST]Well, I'm quite partial to a good squeeze of fresh lemon juice. It adds just the right amount of zesty flavour to whatever I'm cooking up in the kitchen!</s> [INST] Do you have mayonnaise recipes? [/INST]
-
+```python
+"""<s>[INST] What is your favourite condiment? [/INST]Well, I'm quite partial to a good squeeze of fresh lemon juice. It adds just the right amount of zesty flavour to whatever I'm cooking up in the kitchen!</s> [INST] Do you have mayonnaise recipes? [/INST]"""
+```
 </details>
 
 ## Training Hyperparameters