list-of-demos / constants.py
hysts's picture
hysts HF staff
Migrate from yapf to black
85576c8
from huggingface_hub import HfApi
STATUS_CHOICES = [
"RUNNING",
"PAUSED",
"STOPPED",
"RUNTIME_ERROR",
"BUILD_ERROR",
"CONFIG_ERROR",
"BUILDING",
"RUNNING_BUILDING",
"NO_APP_FILE",
]
HARDWARE_CHOICES = [
"cpu-basic",
"cpu-upgrade",
"t4-small",
"t4-medium",
"zero-a10g",
"a10g-small",
"a10g-large",
"a100-large",
]
SDK_CHOICES = [
"gradio",
"streamlit",
"docker",
]
SLEEP_TIME_INT_TO_STR = {
0: "null",
300: "5 minutes",
900: "15 minutes",
1800: "30 minutes",
3600: "1 hour",
36000: "10 hours",
86400: "24 hours",
172800: "48 hours",
259200: "72 hours",
604800: "1 week",
}
SLEEP_TIME_CHOICES = list(SLEEP_TIME_INT_TO_STR.values())
SLEEP_TIME_STR_TO_INT = {v: k for k, v in SLEEP_TIME_INT_TO_STR.items()}
VISIBILITY_CHOICES = ["public", "private"]
api = HfApi()
WHOAMI = api.whoami()["name"]
OWNER_CHOICES = [WHOAMI, "other organizations"]