Spaces:
Paused
Paused
debug
Browse files
main.py
CHANGED
|
@@ -1162,7 +1162,9 @@ def run_generate_frame(images, target_poses, train_steps=100, inference_steps=10
|
|
| 1162 |
is_app=True
|
| 1163 |
|
| 1164 |
print(target_poses)
|
|
|
|
| 1165 |
target_poses = [draw_openpose(pose, height=img_height, width=img_width, include_hands=True, include_face=False) for pose in target_poses]
|
|
|
|
| 1166 |
|
| 1167 |
dwpose, rembg_session, pcdms_model, noise_scheduler, image_encoder_p, image_encoder_g, vae, unet = load_models()
|
| 1168 |
|
|
|
|
| 1162 |
is_app=True
|
| 1163 |
|
| 1164 |
print(target_poses)
|
| 1165 |
+
target_poses = json.loads(target_poses)
|
| 1166 |
target_poses = [draw_openpose(pose, height=img_height, width=img_width, include_hands=True, include_face=False) for pose in target_poses]
|
| 1167 |
+
target_poses[0].save('test.png')
|
| 1168 |
|
| 1169 |
dwpose, rembg_session, pcdms_model, noise_scheduler, image_encoder_p, image_encoder_g, vae, unet = load_models()
|
| 1170 |
|