wenbemi commited on
Commit
e43e95f
Β·
verified Β·
1 Parent(s): 2b87337

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +26 -13
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
- os.environ["HOME"] = str(APP_DIR)
8
- CONFIG_DIR = APP_DIR / ".streamlit"
9
- CONFIG_DIR.mkdir(parents=True, exist_ok=True)
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
- CACHE_ROOT = os.environ.get("HF_CACHE_ROOT") or f"/tmp/hf-cache/u{os.getuid()}"
15
- HF_HOME = pathlib.Path(CACHE_ROOT) / "hf-home"
16
- TRANSFORMERS_CACHE = pathlib.Path(CACHE_ROOT) / "hf-cache"
17
- HUB_CACHE = pathlib.Path(CACHE_ROOT) / "hf-cache"
18
- TORCH_HOME = pathlib.Path(CACHE_ROOT) / "torch-cache"
19
- XDG_CACHE_HOME = pathlib.Path(CACHE_ROOT) / "xdg-cache"
20
-
21
- for p in [HF_HOME, TRANSFORMERS_CACHE, HUB_CACHE, TORCH_HOME, XDG_CACHE_HOME]:
22
- p.mkdir(parents=True, exist_ok=True)
 
 
 
 
 
 
 
 
 
 
 
 
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)