Spaces:
Runtime error
Runtime error
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 |
)
|