mayuema commited on
Commit
d06f4b9
1 Parent(s): f428d8f

first release

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. inference_followyourpose.py +2 -2
app.py CHANGED
@@ -11,6 +11,7 @@ from inference_followyourpose import merge_config_then_run
11
 
12
  HF_TOKEN = os.getenv('HF_TOKEN')
13
  pipe = merge_config_then_run()
 
14
 
15
  with gr.Blocks(css='style.css') as demo:
16
  gr.HTML(
 
11
 
12
  HF_TOKEN = os.getenv('HF_TOKEN')
13
  pipe = merge_config_then_run()
14
+ pipe.download_model()
15
 
16
  with gr.Blocks(css='style.css') as demo:
17
  gr.HTML(
inference_followyourpose.py CHANGED
@@ -19,7 +19,7 @@ class merge_config_then_run():
19
  self.vae = None
20
  self.unet = None
21
 
22
- def download_model():
23
  REPO_ID = 'YueMafighting/FollowYourPose_v1'
24
  snapshot_download(repo_id=REPO_ID, local_dir='./FollowYourPose/checkpoints', local_dir_use_symlinks=False)
25
 
@@ -39,7 +39,7 @@ class merge_config_then_run():
39
  top_crop=0,
40
  bottom_crop=0,
41
  ):
42
- self.download_model()
43
  default_edit_config='./FollowYourPose/configs/pose_sample.yaml'
44
  Omegadict_default_edit_config = OmegaConf.load(default_edit_config)
45
 
 
19
  self.vae = None
20
  self.unet = None
21
 
22
+ def download_model(self):
23
  REPO_ID = 'YueMafighting/FollowYourPose_v1'
24
  snapshot_download(repo_id=REPO_ID, local_dir='./FollowYourPose/checkpoints', local_dir_use_symlinks=False)
25
 
 
39
  top_crop=0,
40
  bottom_crop=0,
41
  ):
42
+
43
  default_edit_config='./FollowYourPose/configs/pose_sample.yaml'
44
  Omegadict_default_edit_config = OmegaConf.load(default_edit_config)
45