Update app.py

app.py

@@ -259,7 +259,8 @@ with gr.Blocks(css=css) as demo:
             image = gr.Image(label="Image Reference", sources=["upload"], type="filepath")
 
             with gr.Column():
-                if sleep_time > 1:
+                current_sleep_time = gr.Number(visible=False)
+                if current_sleep_time > 1:
                     gr.HTML(f'''
                     <div class="gr-prose">
                     <h2>in order to train properly your B-LoRa, you need to set space sleep time to "Don't sleep"</h2>
@@ -285,7 +286,7 @@ with gr.Blocks(css=css) as demo:
                     checkpoint_step = gr.Number(label="checkpoint step", visible=False, value=500)
 
                     
-        if sleep_time > 1:
+        if current_sleep_time > 1:
             train_btn = gr.Button("Train B-LoRa", visible=False)
         else:
             train_btn = gr.Button("Train B-LoRa", visible=True)
@@ -303,5 +304,5 @@ with gr.Blocks(css=css) as demo:
         inputs = [image, b_lora_name, instance_prompt, class_prompt, training_type, training_steps],
         outputs = [status]
     )
-
+demo.load(fn=get_sleep_time, inputs=None, outputs[current_sleep_time])
 demo.launch(debug=True)