Update app.py

app.py

@@ -261,7 +261,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}")
-                if sleep_time > 1:
+                if sleep_time > 1 and sleep_time is not None:
                     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 +287,7 @@ with gr.Blocks(css=css) as demo:
                     checkpoint_step = gr.Number(label="checkpoint step", visible=False, value=500)
 
                     
-        if sleep_time > 1:
+        if sleep_time > 1 and sleep_time is not None:
             train_btn = gr.Button("Train B-LoRa", visible=False)
         else:
             train_btn = gr.Button("Train B-LoRa", visible=True)