shimu0215 commited on
Commit
89d0eac
·
verified ·
1 Parent(s): 8d9bcdc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ class Config:
29
  class ModelManager:
30
  @staticmethod
31
  def load_model(checkpoint_name: str, type='depth'):
32
- if checkpoint_name == 'seg':
33
  model_path = hf_hub_download(
34
  repo_id="shimu0215/seg", # 你的模型仓库
35
  filename="sapiens_1b_seg_foreground_epoch_8_torchscript.pt2", # 你的模型文件
 
29
  class ModelManager:
30
  @staticmethod
31
  def load_model(checkpoint_name: str, type='depth'):
32
+ if type == 'seg':
33
  model_path = hf_hub_download(
34
  repo_id="shimu0215/seg", # 你的模型仓库
35
  filename="sapiens_1b_seg_foreground_epoch_8_torchscript.pt2", # 你的模型文件