wenbemi commited on
Commit
a300d1b
·
verified ·
1 Parent(s): 80b6b64

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -0
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