Spaces:
Sleeping
Sleeping
wenbemi
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,25 +1,38 @@
|
|
| 1 |
# -*- coding: utf-8 -*-
|
| 2 |
# ββββββββββββββββββββββββββββββββ BOOTSTRAP (must be first) ββββββββββββββββββββββββββββββββ
|
| 3 |
import os, pathlib, io, json, random
|
|
|
|
|
|
|
| 4 |
APP_DIR = pathlib.Path(__file__).parent.resolve()
|
| 5 |
|
| 6 |
# Streamlit ν/μ€μ
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
|
| 10 |
-
os.environ["STREAMLIT_HOME"] = str(CONFIG_DIR)
|
| 11 |
os.environ["STREAMLIT_SERVER_HEADLESS"] = "true"
|
| 12 |
os.environ["STREAMLIT_BROWSER_GATHER_USAGE_STATS"] = "false"
|
| 13 |
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
| 18 |
-
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
|
| 24 |
os.environ["HF_HOME"] = str(HF_HOME)
|
| 25 |
os.environ["TRANSFORMERS_CACHE"] = str(TRANSFORMERS_CACHE)
|
|
|
|
| 1 |
# -*- coding: utf-8 -*-
|
| 2 |
# ββββββββββββββββββββββββββββββββ BOOTSTRAP (must be first) ββββββββββββββββββββββββββββββββ
|
| 3 |
import os, pathlib, io, json, random
|
| 4 |
+
|
| 5 |
+
HOME = pathlib.Path.home() # β
μ€ν μ¬μ©μ ν λλ ν°λ¦¬ (μ°κΈ° κ°λ₯)
|
| 6 |
APP_DIR = pathlib.Path(__file__).parent.resolve()
|
| 7 |
|
| 8 |
# Streamlit ν/μ€μ
|
| 9 |
+
STREAMLIT_DIR = HOME / ".streamlit"
|
| 10 |
+
STREAMLIT_DIR.mkdir(parents=True, exist_ok=True)
|
| 11 |
+
os.environ["STREAMLIT_HOME"] = str(STREAMLIT_DIR)
|
|
|
|
| 12 |
os.environ["STREAMLIT_SERVER_HEADLESS"] = "true"
|
| 13 |
os.environ["STREAMLIT_BROWSER_GATHER_USAGE_STATS"] = "false"
|
| 14 |
|
| 15 |
+
# β
HF/Transformers μΊμ: ν λ°μ .cache μ¬μ© (νμ μ HF_CACHE_ROOTλ‘ μ€λ²λΌμ΄λ κ°λ₯)
|
| 16 |
+
CACHE_ROOT = pathlib.Path(os.environ.get("HF_CACHE_ROOT", HOME / ".cache" / f"u{os.getuid()}"))
|
| 17 |
+
HF_HOME = CACHE_ROOT / "hf-home"
|
| 18 |
+
TRANSFORMERS_CACHE = CACHE_ROOT / "hf-cache"
|
| 19 |
+
HUB_CACHE = CACHE_ROOT / "hf-cache"
|
| 20 |
+
TORCH_HOME = CACHE_ROOT / "torch-cache"
|
| 21 |
+
XDG_CACHE_HOME = CACHE_ROOT / "xdg-cache"
|
| 22 |
+
|
| 23 |
+
# ν΄λ μμ± (κΆν μ€λ₯κ° λλ©΄ /tmpλ‘ μλ ν΄λ°±)
|
| 24 |
+
try:
|
| 25 |
+
for p in [HF_HOME, TRANSFORMERS_CACHE, HUB_CACHE, TORCH_HOME, XDG_CACHE_HOME]:
|
| 26 |
+
p.mkdir(parents=True, exist_ok=True)
|
| 27 |
+
except PermissionError:
|
| 28 |
+
TMP_ROOT = pathlib.Path("/tmp") / f"hf-cache-u{os.getuid()}"
|
| 29 |
+
HF_HOME = TMP_ROOT / "hf-home"
|
| 30 |
+
TRANSFORMERS_CACHE = TMP_ROOT / "hf-cache"
|
| 31 |
+
HUB_CACHE = TMP_ROOT / "hf-cache"
|
| 32 |
+
TORCH_HOME = TMP_ROOT / "torch-cache"
|
| 33 |
+
XDG_CACHE_HOME = TMP_ROOT / "xdg-cache"
|
| 34 |
+
for p in [HF_HOME, TRANSFORMERS_CACHE, HUB_CACHE, TORCH_HOME, XDG_CACHE_HOME]:
|
| 35 |
+
p.mkdir(parents=True, exist_ok=True)
|
| 36 |
|
| 37 |
os.environ["HF_HOME"] = str(HF_HOME)
|
| 38 |
os.environ["TRANSFORMERS_CACHE"] = str(TRANSFORMERS_CACHE)
|