Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
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", # 你的模型文件
|