acmyu commited on
Commit
5cb47ee
·
1 Parent(s): 7af14d8
Files changed (1) hide show
  1. main.py +2 -0
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