0Tick commited on
Commit
710be07
·
1 Parent(s): 05a4e8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -68,7 +68,7 @@ def model_selection_changed(model_name):
68
 
69
  def generate(id_task, model_name, batch_count, batch_size, text, *args):
70
  job_count = batch_count
71
-
72
  if current.name != model_name:
73
  current.tokenizer = None
74
  current.model = None
@@ -154,7 +154,6 @@ with gr.Blocks(analytics_enabled=False) as space:
154
 
155
  submit.click(
156
  fn=generate,
157
- _js="submit_promptgen",
158
  inputs=[model_selection, model_selection, batch_count, batch_size, prompt, min_length, max_length, num_beams, temperature, repetition_penalty, length_penalty, sampling_mode, top_k, top_p, ],
159
  outputs=[res, res_info]
160
  )
 
68
 
69
  def generate(id_task, model_name, batch_count, batch_size, text, *args):
70
  job_count = batch_count
71
+ print(model_name,batch_count,batch_size,text)
72
  if current.name != model_name:
73
  current.tokenizer = None
74
  current.model = None
 
154
 
155
  submit.click(
156
  fn=generate,
 
157
  inputs=[model_selection, model_selection, batch_count, batch_size, prompt, min_length, max_length, num_beams, temperature, repetition_penalty, length_penalty, sampling_mode, top_k, top_p, ],
158
  outputs=[res, res_info]
159
  )