Update app.py

app.py

@@ -276,17 +276,19 @@ with gr.Blocks(css=css) as demo:
                         class_prompt = gr.Textbox(label="Specify class prompt", placeholder="style | person | dog ")
                     training_steps = gr.Number(label="Training steps", value=1000, interactive=False)
                     checkpoint_step = gr.Number(label="checkpoint step", visible=False, value=500)
+
+                    training_type.change(
+                        fn = change_training_setup,
+                        inputs = [training_type],
+                        outputs = [training_steps, checkpoint_step]
+                    )
         if sleep_time > 1:
             train_btn = gr.Button("Train B-LoRa", visible=False)
         else:
             train_btn = gr.Button("Train B-LoRa", visible=True)
         status = gr.Textbox(label="status")
 
-    training_type.change(
-        fn = change_training_setup,
-        inputs = [training_type],
-        outputs = [training_steps, checkpoint_step]
-    )
+    
     
     train_btn.click(
         fn = main,