Update README.md

README.md

@@ -75,11 +75,14 @@ Based on the original functions of Chatyuan-large-v1, we optimized the model as
 
 ```python
 from transformers import AutoTokenizer, AutoModel
+from transformers import T5Tokenizer, T5ForConditionalGeneration
 import os
+# 加载模型
+
 model_dir='ClueAI/ChatYuan-large-v2'
-tokenizer = AutoTokenizer.from_pretrained(model_dir)
+tokenizer = T5Tokenizer.from_pretrained(model_dir)
 # 速度会受到网络影响
-model = AutoModel.from_pretrained(model_dir, trust_remote_code=True)
+model = T5ForConditionalGeneration.from_pretrained(model_dir, trust_remote_code=True)
 history = []
 print("starting")
 while True: