Update app.py

app.py

@@ -5,7 +5,7 @@ import torch
 import os
 
 
-CHECKPOINT_PATH=f'output_lh_v2/checkpoint-700'
+CHECKPOINT_PATH=f'linghua_pt-20231202-155337-128-1e-2/checkpoint-400'
 tokenizer = AutoTokenizer.from_pretrained("chatglm3-6b", trust_remote_code=True)
 config = AutoConfig.from_pretrained("chatglm3-6b", trust_remote_code=True, pre_seq_len=128)
 model = AutoModel.from_pretrained("chatglm3-6b", config=config, trust_remote_code=True)
@@ -97,13 +97,8 @@ def reset_state():
     return [], [], None
 
 
-response, history = model.chat(tokenizer, "你是谁", history=[])
-response, history = model.chat(tokenizer, "你好呀", history=history)
-response, history = model.chat(tokenizer, "早上好", history=history)
-
-
 with gr.Blocks() as demo:
-    gr.HTML("""<h1 align="center">ChatGLM2-6B</h1>""")
+    gr.HTML("""<h1 align="center">ChatGLM3-6B</h1>""")
 
     chatbot = gr.Chatbot()
     with gr.Row():