github
/
ChatGLM-6B
mirror of https://github.com/THUDM/ChatGLM-6B
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update web_demo.py

pull/155/head
haofanurusai 2023-03-19 17:29:04 +08:00
parent 88ddfdd5dc
commit 0acf1e5851
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
web_demo.py
View File

@ -19,6 +19,7 @@ def predict(input, max_length, top_p, temperature, history=None):
updates.append(gr.update(visible=True, value="用户：" + query)) updates.append(gr.update(visible=True, value="用户：" + query))
updates.append(gr.update(visible=True, value="ChatGLM-6B：" + response)) updates.append(gr.update(visible=True, value="ChatGLM-6B：" + response))
flag = True flag = True
update_pos = 0
for delta, seq, history in model.chat_stream(tokenizer, input, history, max_length=max_length, top_p=top_p, for delta, seq, history in model.chat_stream(tokenizer, input, history, max_length=max_length, top_p=top_p,
temperature=temperature): temperature=temperature):
@ -26,10 +27,11 @@ def predict(input, max_length, top_p, temperature, history=None):
if flag: if flag:
updates.append(gr.update(visible=True, value="用户：" + input)) updates.append(gr.update(visible=True, value="用户：" + input))
updates.append(gr.update(visible=True, value="ChatGLM-6B：" + response)) updates.append(gr.update(visible=True, value="ChatGLM-6B：" + response))
update_pos = len(updates) - 1
flag = False flag = False
else: else:
updates[-2]=gr.update(visible=True, value="用户：" + input) updates[update_pos-1]=gr.update(visible=True, value="用户：" + input)
updates[-1]=gr.update(visible=True, value="ChatGLM-6B：" + response) updates[update_pos]=gr.update(visible=True, value="ChatGLM-6B：" + response)
if len(updates) < MAX_BOXES: if len(updates) < MAX_BOXES:
updates = updates + [gr.Textbox.update(visible=False)] * (MAX_BOXES - len(updates)) updates = updates + [gr.Textbox.update(visible=False)] * (MAX_BOXES - len(updates))
yield [history] + updates yield [history] + updates