starlight-apk/Qwen
1
0
Fork 0
mirror of https://github.com/QwenLM/Qwen.git synced 2026-05-20 16:35:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix bug

Browse Source
This commit is contained in:
JustinLin610
2023-08-17 16:27:31 +08:00
parent c3c61a14b7
commit 21122b8330
1 changed files with 16 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
web_demo.py
View File

@@ -104,35 +104,35 @@ def _parse_text(text):
def _launch_demo(args, model, tokenizer): def _launch_demo(args, model, tokenizer):
task_history = []
def predict(_query, _chatbot): def predict(_query, _chatbot, _task_history):
print("User: " + _parse_text(_query)) print(f"User: {_parse_text(_query)}")
_chatbot.append((_parse_text(_query), "")) _chatbot.append((_parse_text(_query), ""))
full_response = "" full_response = ""
for response in model.chat_stream(tokenizer, _query, history=task_history): for response in model.chat_stream(tokenizer, _query, history=_task_history):
_chatbot[-1] = (_parse_text(_query), _parse_text(response)) _chatbot[-1] = (_parse_text(_query), _parse_text(response))
yield _chatbot yield _chatbot
full_response = _parse_text(response) full_response = _parse_text(response)
task_history.append((_query, full_response)) print(f"History: {_task_history}")
print("Qwen-7B-Chat: " + _parse_text(full_response)) _task_history.append((_query, full_response))
print(f"Qwen-7B-Chat: {_parse_text(full_response)}")
def regenerate(_chatbot): def regenerate(_chatbot, _task_history):
if not task_history: if not _task_history:
yield _chatbot yield _chatbot
return return
item = task_history.pop(-1) item = _task_history.pop(-1)
_chatbot.pop(-1) _chatbot.pop(-1)
yield from predict(item[0], _chatbot) yield from predict(item[0], _chatbot, _task_history)
def reset_user_input(): def reset_user_input():
return gr.update(value="") return gr.update(value="")
def reset_state(): def reset_state(_task_history):
task_history.clear() _task_history.clear()
return [] return []
with gr.Blocks() as demo: with gr.Blocks() as demo:
@@ -153,16 +153,17 @@ Qwen-7B-Chat <a href="https://modelscope.cn/models/qwen/Qwen-7B-Chat/summary">
chatbot = gr.Chatbot(label='Qwen-7B-Chat', elem_classes="control-height") chatbot = gr.Chatbot(label='Qwen-7B-Chat', elem_classes="control-height")
query = gr.Textbox(lines=2, label='Input') query = gr.Textbox(lines=2, label='Input')
task_history = gr.State([])
with gr.Row(): with gr.Row():
empty_btn = gr.Button("🧹 Clear History (ζΈ…ι™€εŽ†ε²)") empty_btn = gr.Button("🧹 Clear History (ζΈ…ι™€εŽ†ε²)")
submit_btn = gr.Button("πŸš€ Submit (发送)") submit_btn = gr.Button("πŸš€ Submit (发送)")
regen_btn = gr.Button("πŸ€”οΈ Regenerate (重试)") regen_btn = gr.Button("πŸ€”οΈ Regenerate (重试)")
submit_btn.click(predict, [query, chatbot], [chatbot], show_progress=True) submit_btn.click(predict, [query, chatbot, task_history], [chatbot], show_progress=True)
submit_btn.click(reset_user_input, [], [query]) submit_btn.click(reset_user_input, [], [query])
empty_btn.click(reset_state, outputs=[chatbot], show_progress=True) empty_btn.click(reset_state, [task_history], outputs=[chatbot], show_progress=True)
regen_btn.click(regenerate, [chatbot], [chatbot], show_progress=True) regen_btn.click(regenerate, [chatbot, task_history], [chatbot], show_progress=True)
gr.Markdown("""\ gr.Markdown("""\
<font size=2>Note: This demo is governed by the original license of Qwen-7B. \ <font size=2>Note: This demo is governed by the original license of Qwen-7B. \