face_kps_test

handler.py

@@ -295,6 +295,7 @@ class EndpointHandler():
         ]
 
         print("Start inference...")
+        face_kps_test = draw_kps(pose_image, face_info['kps'])
 
         self.generator = torch.Generator(device=device).manual_seed(42)
 
@@ -303,7 +304,8 @@ class EndpointHandler():
             prompt=prompt,
             negative_prompt=negative_prompt,
             image_embeds=face_emb,
-            image=control_images,
+            # image=control_images,
+            image=[face_kps_test],
             control_mask=control_mask,
             controlnet_conditioning_scale=self.control_scales,
             num_inference_steps=num_inference_steps,