Update preprocess/humanparsing/run_parsing.py

preprocess/humanparsing/run_parsing.py

@@ -9,8 +9,13 @@ from parsing_api import onnx_inference
 import torch
 
 from huggingface_hub import hf_hub_download
-hf_hub_download(repo_id="levihsu/OOTDiffusion", filename="checkpoints/humanparsing/parsing_atr.onnx", local_dir="./checkpoints/humanparsing")
-hf_hub_download(repo_id="levihsu/OOTDiffusion", filename="checkpoints/humanparsing/parsing_lip.onnx", local_dir="./checkpoints/humanparsing")
+
+hf_hub_download(repo_id="levihsu/OOTDiffusion", 
+                filename="checkpoints/humanparsing/parsing_atr.onnx", 
+                local_dir=os.path.join(Path(__file__).absolute().parents[2].absolute(), 'checkpoints/humanparsing'))
+hf_hub_download(repo_id="levihsu/OOTDiffusion", 
+                filename="checkpoints/humanparsing/parsing_lip.onnx", 
+                local_dir=os.path.join(Path(__file__).absolute().parents[2].absolute(), 'checkpoints/humanparsing'))
 
 class Parsing:
     def __init__(self, gpu_id: int):