feishen29 commited on
Commit
dff87b8
1 Parent(s): 454e8e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -37,7 +37,9 @@ import os
37
  from controlnet_aux import OpenposeDetector
38
  from insightface.app import FaceAnalysis
39
  from insightface.utils import face_align
 
40
 
 
41
 
42
  parser = argparse.ArgumentParser(description='IMAGDressing-v1')
43
  parser.add_argument('--if_ipa', type=bool, default=True)
 
37
  from controlnet_aux import OpenposeDetector
38
  from insightface.app import FaceAnalysis
39
  from insightface.utils import face_align
40
+ from huggingface_hub import hf_hub_download
41
 
42
+ hf_hub_download(repo_id="feishen29/IMAGDressing", filename="ckpt/IMAGDressing-v1_512.pt", local_dir=".")
43
 
44
  parser = argparse.ArgumentParser(description='IMAGDressing-v1')
45
  parser.add_argument('--if_ipa', type=bool, default=True)