Update README.md

README.md

@@ -61,20 +61,16 @@ Here provides a code snippet with `apply_chat_template` to show you how to load
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model_name = "Qwen/Qwen2.5-3B-Instruct"
-
-model = AutoModelForCausalLM.from_pretrained(
-    model_name,
-    torch_dtype="auto",
-    device_map="auto"
-)
+model_name = 'deeptoken/Qwen2.5-3B-RuoZhiBa-Instruct'
+model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype='auto', device_map='auto')
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
-prompt = "Give me a short introduction to large language model."
+prompt = '鸡柳是鸡的哪个部位？'
 messages = [
-    {"role": "system", "content": "You are Qwen, created by Alibaba Cloud. You are a helpful assistant."},
+    {"role": "system", "content": "You are a helpful assistant."},
     {"role": "user", "content": prompt}
 ]
+
 text = tokenizer.apply_chat_template(
     messages,
     tokenize=False,
@@ -91,6 +87,8 @@ generated_ids = [
 ]
 
 response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
+
+print(response)
 ```