test

handler.py

@@ -45,6 +45,7 @@ class EndpointHandler():
         root="./",
         providers=["CPUExecutionProvider"],
         )
+        
         self.app.prepare(ctx_id=0, det_size=(640, 640))
         
         openpose = OpenposeDetector.from_pretrained("lllyasviel/ControlNet")
@@ -288,12 +289,10 @@ class EndpointHandler():
         control_mask[y1:y2, x1:x2] = 255
         control_mask = Image.fromarray(control_mask.astype(np.uint8))
         
-        # control_images = [face_kps] + [
-        #     self.controlnet_map_fn[s](img_controlnet).resize((width, height))
-        #     for s in self.my_controlnet_selection
-        # ]
-        face_kps_test = draw_kps(face_image, face_info['kps'])
-
+        control_images = [face_kps] + [
+            self.controlnet_map_fn[s](img_controlnet).resize((width, height))
+            for s in self.my_controlnet_selection
+        ]
 
         print("Start inference...")
 
@@ -304,7 +303,7 @@ class EndpointHandler():
             prompt=prompt,
             negative_prompt=negative_prompt,
             image_embeds=face_emb,
-            image=face_kps_test,
+            image=control_images,
             control_mask=control_mask,
             controlnet_conditioning_scale=self.control_scales,
             num_inference_steps=num_inference_steps,