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:35:58 +08:00
parent 0473dfb97b
commit a339498980
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
web_demo.py
View File

@ -14,9 +14,9 @@ def predict(input, max_length, top_p, temperature, history=None):
history = [] history = []
flag = True flag = True
response = '' response = ''
updates = []
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):
updates = []
response += delta response += delta
if flag: if flag:
updates.append(gr.update(visible=True, value="用户：" + input)) updates.append(gr.update(visible=True, value="用户：" + input))