Omnibus commited on
Commit
4bc3325
·
verified ·
1 Parent(s): 99cd660

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -207,7 +207,7 @@ def load_html(inp,title):
207
 
208
 
209
 
210
- def generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=1048, top_p=0.95, repetition_penalty=1.0, m_choice):
211
  html_out=""
212
  #main_point[0]=prompt
213
  #print(datetime.datetime.now())
@@ -360,11 +360,11 @@ with gr.Blocks() as app:
360
  hist_out_box=gr.JSON(label="History")
361
 
362
 
363
- client_choice.change(load_models,client_choice,[chat_b])
364
  app.load(load_models,client_choice,[chat_b]).then(load_html,None,html)
365
 
366
- sub_b = submit_b.click(generate, [msg,chatbot],[msg,chatbot,sumbox,sum_out_box,hist_out_box,html,m_choice])
367
- sub_e = msg.submit(generate, [msg, chatbot], [msg, chatbot,sumbox,sum_out_box,hist_out_box,html,m_choice])
368
  stop_b.click(None,None,None, cancels=[sub_b,sub_e])
369
 
370
  app.queue(default_concurrency_limit=20).launch()
 
207
 
208
 
209
 
210
+ def generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=1048, top_p=0.95, repetition_penalty=1.0):
211
  html_out=""
212
  #main_point[0]=prompt
213
  #print(datetime.datetime.now())
 
360
  hist_out_box=gr.JSON(label="History")
361
 
362
 
363
+ client_choice.change(load_models,client_choice,[chat_bot])
364
  app.load(load_models,client_choice,[chat_b]).then(load_html,None,html)
365
 
366
+ sub_b = submit_b.click(generate, [msg,chatbot],[msg,chatbot,sumbox,sum_out_box,hist_out_box,html])
367
+ sub_e = msg.submit(generate, [msg, chatbot], [msg, chatbot,sumbox,sum_out_box,hist_out_box,html])
368
  stop_b.click(None,None,None, cancels=[sub_b,sub_e])
369
 
370
  app.queue(default_concurrency_limit=20).launch()