|
|
|
@ -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) |
|
|
|
|