Update app.py

app.py

@@ -24,6 +24,12 @@ if is_gpu_associated:
     else:
         which_gpu = "CPU"
 
+def change_training_setup(training_type):
+    if training_type == "style" :
+        return 1000, 500
+    elif training_type == "concept" :
+        return 2000, 1000
+
 def train_dreambooth_blora_sdxl(instance_data_dir, b_lora_trained_folder, instance_prompt, max_train_steps, checkpoint_steps):
     
     script_filename = "train_dreambooth_b-lora_sdxl.py"  # Assuming it's in the same folder
@@ -203,8 +209,16 @@ with gr.Blocks(css=css) as demo:
                 training_type = gr.Radio(label="Training type", choices=["style", "concept"], value="style")
                 b_lora_name = gr.Textbox(label="Name your B-LoRa model", placeholder="b_lora_trained_folder")
                 instance_prompt = gr.Textbox(label="Create instance prompt", placeholder="[v42]")
+                training_steps = gr.Number(label="Training steps", value=1000)
+                checkpoint_step = gr.Number(label="checkpoint step", visible=False, value=500)
         train_btn = gr.Button("Train B-LoRa")
         status = gr.Textbox(label="status")
+
+    training_type.changes(
+        fn = change_training_setup,
+        inputs = [training_type],
+        outputs = [training_steps, checkpoint_step]
+    )
     
     train_btn.click(
         fn = main,