Update README.md

README.md

@@ -19,6 +19,7 @@ from transformers import AutoTokenizer, AutoModel
 tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm2-6b-int4", trust_remote_code=True)
 model = AutoModel.from_pretrained("THUDM/chatglm2-6b-int4", trust_remote_code=True).half().cuda()
 model = model.eval()
-response, history = model.chat(tokenizer, "你好", history=[])
+text = "你好"
+response, history = model.chat(tokenizer, text, history=[])
 response
 ```