shimu0215 commited on
Commit
bf44145
·
verified ·
1 Parent(s): 41ee966

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="sapiens_0.3b_goliath_best_goliath_mIoU_7673_epoch_194_torchscript.pt2", # 你的模型文件
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)