Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ class ModelManager:
|
|
35 |
# 下载模型到本地(默认存储在 ~/.cache/huggingface/hub)
|
36 |
model_path = hf_hub_download(
|
37 |
repo_id="shimu0215/seg", # 你的模型仓库
|
38 |
-
filename="
|
39 |
)
|
40 |
# checkpoint_path = os.path.join('shimu0215/seg', Config.CHECKPOINTS[checkpoint_name])
|
41 |
model = torch.jit.load(model_path)
|
|
|
35 |
# 下载模型到本地(默认存储在 ~/.cache/huggingface/hub)
|
36 |
model_path = hf_hub_download(
|
37 |
repo_id="shimu0215/seg", # 你的模型仓库
|
38 |
+
filename="sapiens_2b_goliath_best_goliath_mIoU_8131_epoch_200_torchscript.pt2", # 你的模型文件
|
39 |
)
|
40 |
# checkpoint_path = os.path.join('shimu0215/seg', Config.CHECKPOINTS[checkpoint_name])
|
41 |
model = torch.jit.load(model_path)
|