fffiloni commited on
Commit
dc3182d
1 Parent(s): 4c31f1c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -129,8 +129,8 @@ def main(dataset_id,
129
  raise gr.Error("You forgot to name the output folder for your model")
130
 
131
  sleep_time = get_sleep_time(hf_token)
132
- if sleep_time:
133
- swap_sleep_time(hf_token, -1)
134
 
135
  gr.Warning("Training is ongoing ⌛... You can close this tab if you like or just wait.")
136
  gr.Warning("If you did not check the `Remove GPU After training`, don't forget to remove the GPU attribution after you are done. ")
 
129
  raise gr.Error("You forgot to name the output folder for your model")
130
 
131
  sleep_time = get_sleep_time(hf_token)
132
+ if sleep_time:
133
+ swap_sleep_time(hf_token, -1)
134
 
135
  gr.Warning("Training is ongoing ⌛... You can close this tab if you like or just wait.")
136
  gr.Warning("If you did not check the `Remove GPU After training`, don't forget to remove the GPU attribution after you are done. ")