fix bugs

main.py

@@ -1161,10 +1161,7 @@ def run_generate_frame(images, target_poses, train_steps=100, inference_steps=10
     finetune=True
     is_app=True
 
-    print(target_poses)
-    target_poses = json.loads(target_poses)
-    target_poses = [draw_openpose(pose, height=img_height, width=img_width, include_hands=True, include_face=False) for pose in target_poses]
-    target_poses[0].save('test.png')
+    
     
     dwpose, rembg_session, pcdms_model, noise_scheduler, image_encoder_p, image_encoder_g, vae, unet = load_models()
     
@@ -1175,6 +1172,12 @@ def run_generate_frame(images, target_poses, train_steps=100, inference_steps=10
     in_img = images[0]
     in_pose, _ = get_pose(in_img, dwpose, "in_pose.png")
 
+    print(target_poses)
+    target_poses = json.loads(target_poses)
+    width, height = in_img.size
+    target_poses = [Image.fromarray(draw_openpose(pose, height=height, width=width, include_hands=True, include_face=False)) for pose in target_poses]
+    target_poses[0].save('pose.png')
+
     results = inference(modelId, in_img, in_pose, target_poses, inference_steps, None, vae, unet, image_encoder_p, is_app)
     #urls = save_temp_imgs(results)
     
@@ -1184,6 +1187,8 @@ def run_generate_frame(images, target_poses, train_steps=100, inference_steps=10
     
     print("Done!")
     
+    results[0].save('result.png')
+
     return results, getThumbnails(results)