Update README.md

README.md

@@ -50,7 +50,7 @@ AttributeError: module 'torch.library' has no attribute 'register_fake'
 
 ### Quickstart 
 We use a special RuQwen2ForCausalLM class to work with this model:
-```
+```python
 from transformers import Qwen2ForCausalLM, AutoConfig, AutoTokenizer
 import torch
 
@@ -92,7 +92,7 @@ class RuQwen2ForCausalLM(Qwen2ForCausalLM):
         super().save_pretrained(save_directory, *args, **kwargs)
 ```
 Here provides a code snippet with apply_chat_template to show you how to load the tokenizer and model and how to generate contents.
-```
+```python
 def generate(messages):
   input_ids = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors="pt").to(model.device)
   output = model.generate(input_ids,