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", # 你的模型文件
|