Spaces:
Sleeping
Sleeping
Commit ·
997fa91
1
Parent(s): 31062d4
fix : Permission denied: /.cache
Browse files
app.py
CHANGED
|
@@ -11,6 +11,11 @@ import os
|
|
| 11 |
|
| 12 |
from models.fpn_inception import FPNInception # 你自己的模型類別
|
| 13 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
# =====================
|
| 15 |
# 初始化模型
|
| 16 |
# =====================
|
|
|
|
| 11 |
|
| 12 |
from models.fpn_inception import FPNInception # 你自己的模型類別
|
| 13 |
|
| 14 |
+
os.environ["TORCH_HOME"] = "./.cache"
|
| 15 |
+
os.environ["HF_HOME"] = "./.cache"
|
| 16 |
+
os.environ["TRANSFORMERS_CACHE"] = "./.cache"
|
| 17 |
+
os.makedirs("./.cache", exist_ok=True)
|
| 18 |
+
|
| 19 |
# =====================
|
| 20 |
# 初始化模型
|
| 21 |
# =====================
|