Spaces:
Runtime error
Runtime error
mayuema
commited on
Commit
•
d06f4b9
1
Parent(s):
f428d8f
first release
Browse files- app.py +1 -0
- 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 |
-
|
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 |
|