charleselena commited on
Commit
6c25b07
1 Parent(s): ed6f71c

Update handler.py

Browse files

Updated the control_image to image

Files changed (1) hide show
  1. handler.py +3 -2
handler.py CHANGED
@@ -102,13 +102,14 @@ class EndpointHandler():
102
 
103
  # process image
104
  image = self.decode_base64_image(image)
105
- control_image = CONTROLNET_MAPPING[self.control_type]["hinter"](image)
106
 
107
  # run inference pipeline
108
  out = self.pipe(
109
  prompt=prompt,
110
  negative_prompt=negative_prompt,
111
- image=control_image,
 
112
  num_inference_steps=num_inference_steps,
113
  guidance_scale=guidance_scale,
114
  num_images_per_prompt=1,
 
102
 
103
  # process image
104
  image = self.decode_base64_image(image)
105
+ #control_image = CONTROLNET_MAPPING[self.control_type]["hinter"](image)
106
 
107
  # run inference pipeline
108
  out = self.pipe(
109
  prompt=prompt,
110
  negative_prompt=negative_prompt,
111
+ # image=control_image,
112
+ image=image,
113
  num_inference_steps=num_inference_steps,
114
  guidance_scale=guidance_scale,
115
  num_images_per_prompt=1,