Spaces:
Sleeping
Sleeping
wenbemi
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,6 +1,20 @@
|
|
| 1 |
# streamlit 라이브러리가 import 되기 전에,
|
| 2 |
# 설정 파일 경로를 앱 내부의 쓰기 가능한 경로로 강제 지정합니다.
|
| 3 |
import os, pathlib, io
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
from huggingface_hub import hf_hub_download
|
| 5 |
import pandas as pd
|
| 6 |
import json
|
|
|
|
| 1 |
# streamlit 라이브러리가 import 되기 전에,
|
| 2 |
# 설정 파일 경로를 앱 내부의 쓰기 가능한 경로로 강제 지정합니다.
|
| 3 |
import os, pathlib, io
|
| 4 |
+
APP_DIR = pathlib.Path(__file__).parent.resolve()
|
| 5 |
+
|
| 6 |
+
# 모든 HF/Transformers 캐시를 /tmp 쪽으로 강제
|
| 7 |
+
os.environ.setdefault("HF_HOME", "/tmp/hf-home")
|
| 8 |
+
os.environ.setdefault("TRANSFORMERS_CACHE", "/tmp/hf-cache")
|
| 9 |
+
os.environ.setdefault("HUGGINGFACE_HUB_CACHE", "/tmp/hf-cache")
|
| 10 |
+
os.environ.setdefault("TORCH_HOME", "/tmp/torch-cache")
|
| 11 |
+
os.environ.setdefault("XDG_CACHE_HOME", "/tmp/xdg-cache")
|
| 12 |
+
os.environ.setdefault("TOKENIZERS_PARALLELISM", "false")
|
| 13 |
+
|
| 14 |
+
# 디렉터리 보장
|
| 15 |
+
for p in ["/tmp/hf-home", "/tmp/hf-cache", "/tmp/torch-cache", "/tmp/xdg-cache"]:
|
| 16 |
+
os.makedirs(p, exist_ok=True)
|
| 17 |
+
|
| 18 |
from huggingface_hub import hf_hub_download
|
| 19 |
import pandas as pd
|
| 20 |
import json
|