Update app.py

app.py

@@ -51,6 +51,8 @@ def get_sleep_time(hf_token):
     except:
         gcTimeout = None
     return gcTimeout
+
+sleep_time = get_sleep_time(hf_token)
     
 def train_dreambooth_blora_sdxl(instance_data_dir, b_lora_trained_folder, instance_prompt, class_prompt, max_train_steps, checkpoint_steps):
     
@@ -257,14 +259,27 @@ with gr.Blocks(css=css) as demo:
             image = gr.Image(label="Image Reference", sources=["upload"], type="filepath")
 
             with gr.Column():
-                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")
-                with gr.Row():
-                    instance_prompt = gr.Textbox(label="Create instance prompt", placeholder="A [v42] <class_prompt>")
-                    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)
-        train_btn = gr.Button("Train B-LoRa")
+                if 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>
+                    <p class="actions">
+                        <a href="https://huggingface.co/spaces/{os.environ['SPACE_ID']}/settings">⚡️ &nbsp; Set recommended Sleep Time</a>
+                    </p>
+                    </div>
+                    ''')
+                else:    
+                    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")
+                    with gr.Row():
+                        instance_prompt = gr.Textbox(label="Create instance prompt", placeholder="A [v42] <class_prompt>")
+                        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)
+        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(