From 3aaf9d3456a552c86d3ef8c032a3a6322cd4d3c4 Mon Sep 17 00:00:00 2001
From: Aohan Zeng <Sengxian@live.com>
Date: Tue, 14 Mar 2023 03:06:37 +0800
Subject: [PATCH] Update cli_demo.py

---
 cli_demo.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/cli_demo.py b/cli_demo.py
index 97a248a..07cfd6f 100644
--- a/cli_demo.py
+++ b/cli_demo.py
@@ -6,14 +6,15 @@ model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True).ha
 model = model.eval()
 
 history = []
+print("欢迎使用 ChatGLM-6B 模型，输入内容即可进行对话，clear 清空对话历史，stop 终止程序")
 while True:
-    query = input("请输入提示，clear清空对话历史，stop终止程序：\n")
+    query = input("\n用户：")
     if query == "stop":
         break
     if query == "clear":
         history = []
         os.system('clear')
         continue
-    print("回复：")
     response, history = model.chat(tokenizer, query, history=history)
-    print(response)
+    print(f"ChatGLM-6B：{response}")
+