Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
)
|