Update app.py

app.py

@@ -209,7 +209,7 @@ 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)
+                training_steps = gr.Number(label="Training steps", value=1000, interactive=False)
                 checkpoint_step = gr.Number(label="checkpoint step", visible=False, value=500)
         train_btn = gr.Button("Train B-LoRa")
         status = gr.Textbox(label="status")
@@ -222,7 +222,7 @@ with gr.Blocks(css=css) as demo:
     
     train_btn.click(
         fn = main,
-        inputs = [image, b_lora_name, instance_prompt],
+        inputs = [image, b_lora_name, instance_prompt, training_type],
         outputs = [status]
     )