Spaces:
Running
Running
Update flux_train_ui.py
Browse files- flux_train_ui.py +1 -1
flux_train_ui.py
CHANGED
@@ -406,7 +406,7 @@ with gr.Blocks(theme=theme, css=css) as demo:
|
|
406 |
outputs=[captioning_area, sample, start]
|
407 |
)
|
408 |
|
409 |
-
start.click(fn=create_dataset, inputs=[images]
|
410 |
fn=start_training,
|
411 |
inputs=[
|
412 |
lora_name,
|
|
|
406 |
outputs=[captioning_area, sample, start]
|
407 |
)
|
408 |
|
409 |
+
start.click(fn=create_dataset, inputs=[images], outputs=dataset_folder).then(
|
410 |
fn=start_training,
|
411 |
inputs=[
|
412 |
lora_name,
|