test

handler.py

@@ -40,6 +40,7 @@ class EndpointHandler():
         #     root=f"./antelopev2",
         #     providers=["CPUExecutionProvider"],
         # )
+        
         self.app = FaceAnalysis(
         name="buffalo_l",
         root="./",
@@ -295,7 +296,6 @@ class EndpointHandler():
         ]
 
         print("Start inference...")
-        face_kps_test = draw_kps(pose_image, face_info['kps'])
 
         self.generator = torch.Generator(device=device).manual_seed(42)
 
@@ -304,8 +304,7 @@ class EndpointHandler():
             prompt=prompt,
             negative_prompt=negative_prompt,
             image_embeds=face_emb,
-            # image=control_images,
-            image=[face_kps_test],
+            image=control_images,
             control_mask=control_mask,
             controlnet_conditioning_scale=self.control_scales,
             num_inference_steps=num_inference_steps,