Update app.py

app.py

@@ -271,8 +271,13 @@ with gr.Blocks(css=css) as demo:
             with gr.Column():
                 
                     with gr.Row():
-                        current_sleep_time = gr.Textbox(label="current space sleep time")
-                        check_sleep_time_btn = gr.Button("check", scale=1)
+                        current_sleep_time = gr.Dropdown(
+                            label="current space sleep time",
+                            choices = [
+                                -1, 300, 900, 1800, 3600, 36000, 86400, 172800, 259200, 604800
+                            ]
+                        )
+                        #check_sleep_time_btn = gr.Button("check", scale=1)
                     training_type = gr.Radio(label="Training type", choices=["style", "concept"], value="style", visible=False)
                     b_lora_name = gr.Textbox(label="Name your B-LoRa model", placeholder="b_lora_trained_folder", visible=False)
                     with gr.Row():
@@ -287,10 +292,10 @@ with gr.Blocks(css=css) as demo:
     
         status = gr.Textbox(label="status")
 
-    check_sleep_time_btn.click(
-        fn = check_sleep_time,
-        inputs = None, 
-        outputs = [current_sleep_time, b_lora_name, instance_prompt, class_prompt, training_type, training_steps, train_btn]
+    current_sleep_time.change(
+        fn = swap_sleep_time,
+        inputs = current_sleep_time, 
+        outputs = None
     )
 
     demo.load(
@@ -300,10 +305,10 @@ with gr.Blocks(css=css) as demo:
     )
     
     training_type.change(
-                        fn = change_training_setup,
-                        inputs = [training_type],
-                        outputs = [training_steps, checkpoint_step]
-                    )
+        fn = change_training_setup,
+        inputs = [training_type],
+        outputs = [training_steps, checkpoint_step]
+    )
     
     train_btn.click(
         fn = main,