Browse Source

Add another web demo with Gradio

pull/373/head^2
duzx16 2 years ago
parent
commit
ec069419be
  1. 20
      web_demo3.py

20
web_demo3.py

@ -4,13 +4,12 @@ from typing import TYPE_CHECKING, Any, Callable, Dict, List, Tuple, Type
import mdtex2html
tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True)
model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True).quantize(8).half().cuda()
model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True).half().cuda()
model = model.eval()
# MAX_TURNS = 20
# MAX_BOXES = MAX_TURNS * 2
"""Override Chatbot.postprocess"""
def postprocess(self, y):
if y is None:
return []
@ -20,6 +19,8 @@ def postprocess(self, y):
None if response is None else mdtex2html.convert(response),
)
return y
gr.Chatbot.postprocess = postprocess
@ -30,6 +31,7 @@ def predict(input, chatbot, max_length, top_p, temperature, history):
chatbot[-1] = (input, response)
yield chatbot, history
def reset_user_input():
return gr.update(value='')
@ -37,6 +39,7 @@ def reset_user_input():
def reset_state():
return [], []
with gr.Blocks() as demo:
gr.HTML("""<h1 align="center">ChatGLM</h1>""")
@ -57,13 +60,14 @@ with gr.Blocks() as demo:
history = gr.State([])
user_input.submit(predict, [user_input, chatbot, max_length, top_p, temperature, history], [chatbot, history], show_progress=True)
user_input.submit(predict, [user_input, chatbot, max_length, top_p, temperature, history], [chatbot, history],
show_progress=True)
user_input.submit(reset_user_input, [], [user_input])
submitBtn.click(predict, [user_input, chatbot, max_length, top_p, temperature, history], [chatbot, history], show_progress=True)
submitBtn.click(predict, [user_input, chatbot, max_length, top_p, temperature, history], [chatbot, history],
show_progress=True)
submitBtn.click(reset_user_input, [], [user_input])
emptyBtn.click(reset_state, outputs=[chatbot, history], show_progress=True)
demo.queue().launch(share=False, inbrowser=True)
demo.queue().launch(share=True, inbrowser=True)

Loading…
Cancel
Save