Update app.py

app.py

@@ -129,8 +129,8 @@ def main(dataset_id,
         raise gr.Error("You forgot to name the output folder for your model")
 
     sleep_time = get_sleep_time(hf_token)
-        if sleep_time:
-            swap_sleep_time(hf_token, -1)
+    if sleep_time:
+        swap_sleep_time(hf_token, -1)
 
     gr.Warning("Training is ongoing ⌛... You can close this tab if you like or just wait.")
     gr.Warning("If you did not check the `Remove GPU After training`, don't forget to remove the GPU attribution after you are done. ")