Spaces:
Running
on
Zero
Running
on
Zero
Update preprocess/humanparsing/run_parsing.py
Browse files
preprocess/humanparsing/run_parsing.py
CHANGED
@@ -22,8 +22,8 @@ class Parsing:
|
|
22 |
session_options.add_session_config_entry('gpu_id', str(gpu_id))
|
23 |
|
24 |
parsing_ckpt_path = os.path.join(Path(__file__).absolute().parents[2].absolute(), 'checkpoints/humanparsing')
|
25 |
-
atr_model_path = 'https://huggingface.co/levihsu/OOTDiffusion/
|
26 |
-
lip_model_path = 'https://huggingface.co/levihsu/OOTDiffusion/
|
27 |
|
28 |
from basicsr.utils.download_util import load_file_from_url
|
29 |
load_file_from_url(atr_model_path, model_dir=parsing_ckpt_path)
|
|
|
22 |
session_options.add_session_config_entry('gpu_id', str(gpu_id))
|
23 |
|
24 |
parsing_ckpt_path = os.path.join(Path(__file__).absolute().parents[2].absolute(), 'checkpoints/humanparsing')
|
25 |
+
atr_model_path = 'https://huggingface.co/levihsu/OOTDiffusion/resolve/main/checkpoints/humanparsing/parsing_atr.onnx'
|
26 |
+
lip_model_path = 'https://huggingface.co/levihsu/OOTDiffusion/resolve/main/checkpoints/humanparsing/parsing_lip.onnx'
|
27 |
|
28 |
from basicsr.utils.download_util import load_file_from_url
|
29 |
load_file_from_url(atr_model_path, model_dir=parsing_ckpt_path)
|