diff --git a/web_demo2.py b/web_demo2.py
index 6946a15..9ae6b26 100644
--- a/web_demo2.py
+++ b/web_demo2.py
@@ -11,7 +11,7 @@ st.set_page_config(
 
 @st.cache_resource
 def get_model():
-    tokenizer = AutoTokenizer.from_pretrained("/THUDM/chatglm-6b", trust_remote_code=True)
+    tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True)
     model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True).half().cuda()
     model = model.eval()
     return tokenizer, model
@@ -27,16 +27,10 @@ def predict(input, history=None):
         history = []
     response, history = model.chat(tokenizer, input, history)
 
-    #updates = []
     for i, (query, response) in enumerate(history):
-        #updates.append("用户：" + query)
         message(query, avatar_style="big-smile", key=str(i) + "_user")
-        #updates.append("ChatGLM-6B：" + response)
         message(response, avatar_style="bottts", key=str(i))
 
-    # if len(updates) < MAX_BOXES:
-    #     updates = updates + [""] * (MAX_BOXES - len(updates))
-
     return history