Update app.py

app.py

@@ -4,7 +4,7 @@ import mdtex2html
 from utils import load_model_on_gpus
 
 tokenizer = AutoTokenizer.from_pretrained("IPLab/chatglm2-6b-patent-7.30", trust_remote_code=True)
-model = AutoModel.from_pretrained("IPLab/chatglm2-6b-patent-7.30", trust_remote_code=True).float()
+model = AutoModel.from_pretrained("IPLab/chatglm2-6b-patent-7.30", trust_remote_code=True).cuda()
 # 多显卡支持，使用下面两行代替上面一行，将num_gpus改为你实际的显卡数量
 # from utils import load_model_on_gpus
 # model = load_model_on_gpus("THUDM/chatglm2-6b", num_gpus=2)
@@ -80,7 +80,7 @@ def reset_state():
 
 
 with gr.Blocks() as demo:
-    gr.HTML("""<h1 align="center">青云大模型Beta测试版8.1</h1>""")
+    gr.HTML("""<h1 align="center">青云大模型Beta测试版</h1>""")
 
     chatbot = gr.Chatbot()
     with gr.Row():