Hugging Face's logo

Spaces:
levihsu
/
OOTDiffusion
Running on Zero

App Files Community
32
levihsu commited on
Commit
ebd84dc
·
verified ·
1 Parent(s): 1be901d

Update preprocess/humanparsing/run_parsing.py

Browse files
Files changed (1) hide show
  1. preprocess/humanparsing/run_parsing.py +7 -2
preprocess/humanparsing/run_parsing.py CHANGED
@@ -9,8 +9,13 @@ from parsing_api import onnx_inference
9
  import torch
10
 
11
  from huggingface_hub import hf_hub_download
12
- hf_hub_download(repo_id="levihsu/OOTDiffusion", filename="checkpoints/humanparsing/parsing_atr.onnx", local_dir="./checkpoints/humanparsing")
13
- hf_hub_download(repo_id="levihsu/OOTDiffusion", filename="checkpoints/humanparsing/parsing_lip.onnx", local_dir="./checkpoints/humanparsing")
 
 
 
 
 
14
 
15
  class Parsing:
16
  def __init__(self, gpu_id: int):
 
9
  import torch
10
 
11
  from huggingface_hub import hf_hub_download
12
+
13
+ hf_hub_download(repo_id="levihsu/OOTDiffusion",
14
+ filename="checkpoints/humanparsing/parsing_atr.onnx",
15
+ local_dir=os.path.join(Path(__file__).absolute().parents[2].absolute(), 'checkpoints/humanparsing'))
16
+ hf_hub_download(repo_id="levihsu/OOTDiffusion",
17
+ filename="checkpoints/humanparsing/parsing_lip.onnx",
18
+ local_dir=os.path.join(Path(__file__).absolute().parents[2].absolute(), 'checkpoints/humanparsing'))
19
 
20
  class Parsing:
21
  def __init__(self, gpu_id: int):