Improve the chat prompt explanation and example

README.md

@@ -61,11 +61,22 @@ Each message should be a dict with two keys:
 
 For example:
 ```python
+from transformers import AutoTokenizer
+
+tokenizer = AutoTokenizer.from_pretrained("mathewhe/DCLM-7B-Chat")
+
 messages = [
     {"role": "user", "content": "Solve for x: 3x=4"},
     {"role": "assistant", "content": "3x=4\n(3x)/3=(4)/3\nx=4/3"},
     {"role": "user", "content": "Please explain your work."},
 ]
+print(tokenizer.apply_chat_template(messages, tokenize=False)
+```
+outputs
+```
+<|endoftext|>[INST] Solve for x: 3x=4 [/INST][ASST] 3x=4
+(3x)/3=(4)/3
+x=4/3 [/ASST]<|endoftext|><|endoftext|>[INST] Please explain your work [/INST]
 ```
 
 See the example code in the included `chat_class.py` module for more details.