Spaces:
Runtime error
Runtime error
Matthias Kleiner
commited on
Commit
·
1f0d187
1
Parent(s):
9652e54
added wait and retry if private space needed to be restarted
Browse files
app.py
CHANGED
|
@@ -4,6 +4,7 @@ import os
|
|
| 4 |
import random
|
| 5 |
import datetime
|
| 6 |
from huggingface_hub import hf_api
|
|
|
|
| 7 |
|
| 8 |
def check_password(username, password):
|
| 9 |
if password == os.environ["ACCESS"]:
|
|
@@ -24,6 +25,21 @@ def func(file, number_of_pages, secret):
|
|
| 24 |
space_runtime_after_restart = hf_api.restart_space("ByMatthew/deckify_private", token=read_key)
|
| 25 |
print(f"Space runtime after restart: {space_runtime_after_restart}")
|
| 26 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
client = Client("ByMatthew/deckify_private", hf_token=read_key)
|
| 28 |
|
| 29 |
print(f"Client: {client}")
|
|
|
|
| 4 |
import random
|
| 5 |
import datetime
|
| 6 |
from huggingface_hub import hf_api
|
| 7 |
+
import time
|
| 8 |
|
| 9 |
def check_password(username, password):
|
| 10 |
if password == os.environ["ACCESS"]:
|
|
|
|
| 25 |
space_runtime_after_restart = hf_api.restart_space("ByMatthew/deckify_private", token=read_key)
|
| 26 |
print(f"Space runtime after restart: {space_runtime_after_restart}")
|
| 27 |
|
| 28 |
+
|
| 29 |
+
max_retries = 20
|
| 30 |
+
retry_delay = 10
|
| 31 |
+
success = False
|
| 32 |
+
for i in range(max_retries):
|
| 33 |
+
space_runtime = hf_api.get_space_runtime("ByMatthew/deckify_private", token=read_key)
|
| 34 |
+
print(f"Space runtime: {space_runtime}")
|
| 35 |
+
if space_runtime.stage == "RUNNING":
|
| 36 |
+
success = True
|
| 37 |
+
break
|
| 38 |
+
time.sleep(retry_delay)
|
| 39 |
+
|
| 40 |
+
if not success:
|
| 41 |
+
return "Failed to start the private space in time. Please try again later."
|
| 42 |
+
|
| 43 |
client = Client("ByMatthew/deckify_private", hf_token=read_key)
|
| 44 |
|
| 45 |
print(f"Client: {client}")
|