Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -226,35 +226,23 @@ Due to resource limitation, you can only train 10 epochs on maximum.
|
|
226 |
)
|
227 |
|
228 |
with gr.Row():
|
229 |
-
|
230 |
-
|
231 |
-
|
232 |
-
|
233 |
-
|
234 |
-
|
235 |
-
|
236 |
-
|
237 |
-
|
238 |
-
|
239 |
-
|
240 |
-
|
241 |
-
|
242 |
-
|
243 |
-
|
244 |
-
|
245 |
-
|
246 |
-
task,
|
247 |
-
tempdir_path,
|
248 |
-
log_every,
|
249 |
-
max_epoch,
|
250 |
-
scheduler,
|
251 |
-
warmup_steps,
|
252 |
-
optimizer,
|
253 |
-
learning_rate,
|
254 |
-
weight_decay,
|
255 |
-
],
|
256 |
-
[trained_model, hyp_text]
|
257 |
-
)
|
258 |
|
259 |
gr.Markdown(load_markdown())
|
260 |
|
|
|
226 |
)
|
227 |
|
228 |
with gr.Row():
|
229 |
+
finetune_btn = gr.Button("Finetune Model", variant="primary")
|
230 |
+
finetune_btn.click(
|
231 |
+
finetune_model,
|
232 |
+
[
|
233 |
+
lang,
|
234 |
+
task,
|
235 |
+
tempdir_path,
|
236 |
+
log_every,
|
237 |
+
max_epoch,
|
238 |
+
scheduler,
|
239 |
+
warmup_steps,
|
240 |
+
optimizer,
|
241 |
+
learning_rate,
|
242 |
+
weight_decay,
|
243 |
+
],
|
244 |
+
[trained_model, hyp_text]
|
245 |
+
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
246 |
|
247 |
gr.Markdown(load_markdown())
|
248 |
|