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

Add markdown for web demo

pull/9/head
duzx16 2023-03-14 14:55:49 +08:00
parent 9350b81709
commit 6113d4c02f
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
web_demo.py
View File

@ -13,8 +13,8 @@ def predict(input, history=[]):
response, history = model.chat(tokenizer, input, history) response, history = model.chat(tokenizer, input, history)
updates = [] updates = []
for query, response in history: for query, response in history:
updates.append(gr.update(visible=True, value=query)) updates.append(gr.update(visible=True, value="用户：" + query))
updates.append(gr.update(visible=True, value=response)) updates.append(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))
return [history] + updates return [history] + updates
@ -25,10 +25,9 @@ with gr.Blocks() as demo:
text_boxes = [] text_boxes = []
for i in range(MAX_BOXES): for i in range(MAX_BOXES):
if i % 2 == 0: if i % 2 == 0:
label = "提问：" text_boxes.append(gr.Markdown(visible=False, label="提问："))
else: else:
label = "回复：" text_boxes.append(gr.Markdown(visible=False, label="回复："))
text_boxes.append(gr.Textbox(visible=False, label=label))
with gr.Row(): with gr.Row():
with gr.Column(scale=4): with gr.Column(scale=4):