Update app.py

app.py

@@ -50,9 +50,10 @@ def get_sleep_time(hf_token):
         gcTimeout = response.json()['runtime']['gcTimeout']
     except:
         gcTimeout = None
+    sleep_time = gcTimeout
     return gcTimeout
 
-sleep_time = get_sleep_time(hf_token)
+#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):
     
@@ -261,7 +262,7 @@ with gr.Blocks(css=css) as demo:
             with gr.Column():
                 current_sleep_time = gr.Number(visible=False)
                 print(f"loaded demo: {current_sleep_time['value']}")
-                if sleep_time > 1 and sleep_time is not None:
+                if sleep_time is not None and 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>
@@ -287,7 +288,7 @@ with gr.Blocks(css=css) as demo:
                     checkpoint_step = gr.Number(label="checkpoint step", visible=False, value=500)
 
                     
-        if sleep_time > 1 and sleep_time is not None:
+        if sleep_time is not None and sleep_time > 1:
             train_btn = gr.Button("Train B-LoRa", visible=False)
         else:
             train_btn = gr.Button("Train B-LoRa", visible=True)