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 16:45:40 +08:00
parent a339498980
commit 649e3f280a
1 changed files with 3 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
web_demo.py
View File

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