Update README.md

README.md

@@ -305,7 +305,7 @@ We included GPT-3.5, GPT-4, LLaMA2-7B-Base, ChatMusician-Base, and ChatMusician.
 You can use the models through Huggingface's Transformers library. Check our Github repo for more advanced use: [https://github.com/hf-lin/ChatMusician](https://github.com/hf-lin/ChatMusician)
 
 ## CLI demo
-```
+```python
 from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
 from string import Template
 prompt_template = Template("Human: ${inst} </s> Assistant: ")