Jacobmadwed commited on
Commit
016f972
·
verified ·
1 Parent(s): 0b8c57d

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -5
handler.py CHANGED
@@ -9,12 +9,9 @@ import diffusers
9
  from diffusers.utils import load_image
10
  from diffusers.models import ControlNetModel
11
  from diffusers.pipelines.controlnet.multicontrolnet import MultiControlNetModel
12
- from insightface.app import FaceAnalysis
13
  from style_template import styles
14
  from pipeline_stable_diffusion_xl_instantid_full import StableDiffusionXLInstantIDPipeline, draw_kps
15
  from controlnet_aux import OpenposeDetector
16
- import torch.nn.functional as F
17
- from torchvision.transforms import Compose
18
  import os
19
  from huggingface_hub import hf_hub_download
20
  import base64
@@ -240,7 +237,7 @@ class EndpointHandler:
240
 
241
  pose_image_base64 = data.get("pose_image_base64")
242
  pose_image = None
243
- if (pose_image_base64):
244
  pose_image_data = base64.b64decode(pose_image_base64)
245
  pose_image = Image.open(io.BytesIO(pose_image_data))
246
 
@@ -300,7 +297,7 @@ class EndpointHandler:
300
  outputs = self.pipe(
301
  prompt=inputs,
302
  negative_prompt=negative_prompt,
303
- image_embeds=face_emb,
304
  image=control_images,
305
  control_mask=control_mask,
306
  controlnet_conditioning_scale=control_scales,
 
9
  from diffusers.utils import load_image
10
  from diffusers.models import ControlNetModel
11
  from diffusers.pipelines.controlnet.multicontrolnet import MultiControlNetModel
 
12
  from style_template import styles
13
  from pipeline_stable_diffusion_xl_instantid_full import StableDiffusionXLInstantIDPipeline, draw_kps
14
  from controlnet_aux import OpenposeDetector
 
 
15
  import os
16
  from huggingface_hub import hf_hub_download
17
  import base64
 
237
 
238
  pose_image_base64 = data.get("pose_image_base64")
239
  pose_image = None
240
+ if pose_image_base64:
241
  pose_image_data = base64.b64decode(pose_image_base64)
242
  pose_image = Image.open(io.BytesIO(pose_image_data))
243
 
 
297
  outputs = self.pipe(
298
  prompt=inputs,
299
  negative_prompt=negative_prompt,
300
+ image_embeds=None, # Removed face_emb
301
  image=control_images,
302
  control_mask=control_mask,
303
  controlnet_conditioning_scale=control_scales,