mikeee commited on
Commit
23c8307
1 Parent(s): 55c435d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -8,6 +8,7 @@
8
  from textwrap import dedent
9
 
10
  # credit to https://github.com/THUDM/ChatGLM2-6B/blob/main/web_demo.py
 
11
  from transformers import AutoModel, AutoTokenizer
12
  import gradio as gr
13
  import mdtex2html
@@ -166,6 +167,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
166
  history = gr.State([])
167
  past_key_values = gr.State(None)
168
 
 
 
169
  submitBtn.click(predict, [user_input, chatbot, max_length, top_p, temperature, history, past_key_values],
170
  [chatbot, history, past_key_values], show_progress=True, api_name="predict")
171
  submitBtn.click(reset_user_input, [], [user_input])
 
8
  from textwrap import dedent
9
 
10
  # credit to https://github.com/THUDM/ChatGLM2-6B/blob/main/web_demo.py
11
+ # while mistakes are mine
12
  from transformers import AutoModel, AutoTokenizer
13
  import gradio as gr
14
  import mdtex2html
 
167
  history = gr.State([])
168
  past_key_values = gr.State(None)
169
 
170
+ user_input.submit(predict, [user_input, chatbot, max_length, top_p, temperature, history, past_key_values],
171
+ [chatbot, history, past_key_values], show_progress=True)
172
  submitBtn.click(predict, [user_input, chatbot, max_length, top_p, temperature, history, past_key_values],
173
  [chatbot, history, past_key_values], show_progress=True, api_name="predict")
174
  submitBtn.click(reset_user_input, [], [user_input])