Update flux_train_ui.py

flux_train_ui.py

@@ -389,8 +389,7 @@ with gr.Blocks(theme=theme, css=css) as demo:
     dataset_folder = gr.State()
 
     images.upload(
-        #load_captioning,
-        None,
+        load_captioning,
         inputs=[images, concept_sentence],
         outputs=output_components
     )
@@ -406,7 +405,7 @@ with gr.Blocks(theme=theme, css=css) as demo:
         outputs=[captioning_area, sample, start]
     )
     
-    start.click(fn=create_dataset, inputs=[images], outputs=dataset_folder).then(
+    start.click(fn=create_dataset, inputs=[images] + caption_list, outputs=dataset_folder).then(
         fn=start_training,
         inputs=[
             lora_name,
@@ -426,7 +425,7 @@ with gr.Blocks(theme=theme, css=css) as demo:
         outputs=progress_area,
     )
 
-    do_captioning.click(fn=run_captioning, inputs=[images, concept_sentence], outputs=caption_list)
+    do_captioning.click(fn=run_captioning, inputs=[images, concept_sentence] + caption_list, outputs=caption_list)
 
 if __name__ == "__main__":
     demo.launch(share=True, show_error=True)