Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,7 +5,8 @@ from huggingface_hub import hf_hub_download
|
|
| 5 |
model_path = hf_hub_download(
|
| 6 |
repo_id="Hanxiaofeng123/Deepcube", # 替换成你自己的仓库
|
| 7 |
repo_type="space",
|
| 8 |
-
filename="checkpoint/final_model_K_30.pth"
|
|
|
|
| 9 |
)
|
| 10 |
|
| 11 |
import flask
|
|
|
|
| 5 |
model_path = hf_hub_download(
|
| 6 |
repo_id="Hanxiaofeng123/Deepcube", # 替换成你自己的仓库
|
| 7 |
repo_type="space",
|
| 8 |
+
filename="checkpoint/final_model_K_30.pth",
|
| 9 |
+
cache_dir="/app/.cache" # 指定在工作目录下缓存
|
| 10 |
)
|
| 11 |
|
| 12 |
import flask
|