|
|
|
@ -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 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|