Spaces:

xiangxiang
/

chatglm2-6b

Running

xiangxiang commited on Aug 4, 2023

Commit

ca874bf

•

1 Parent(s): 07619d2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,8 +2,8 @@ from transformers import AutoModel, AutoTokenizer
 import gradio as gr
 import mdtex2html
-tokenizer = AutoTokenizer.from_pretrained("xiangxiang/chatglm2-6b-WaJiaBank", trust_remote_code=True)
-model = AutoModel.from_pretrained("xiangxiang/chatglm2-6b-WaJiaBank",trust_remote_code=True)
 model = model.eval()
 """Override Chatbot.postprocess"""

 import gradio as gr
 import mdtex2html
+tokenizer = AutoTokenizer.from_pretrained("THUDM/ChatGLM-6B", trust_remote_code=True)
+model = AutoModel.from_pretrained("THUDM/ChatGLM-6B",trust_remote_code=True)
 model = model.eval()
 """Override Chatbot.postprocess"""