Update handler.py
Browse files- handler.py +1 -2
handler.py
CHANGED
@@ -124,7 +124,6 @@ class EndpointHandler():
|
|
124 |
image=control_image,
|
125 |
num_inference_steps=num_inference_steps,
|
126 |
guidance_scale=guidance_scale,
|
127 |
-
num_images_per_prompt=4,
|
128 |
height=height,
|
129 |
width=width,
|
130 |
controlnet_conditioning_scale=controlnet_conditioning_scale,
|
@@ -133,7 +132,7 @@ class EndpointHandler():
|
|
133 |
|
134 |
# generator=self.generator COMMENTED from self.pipe
|
135 |
# return the first generated PIL image
|
136 |
-
return out.images
|
137 |
|
138 |
# helper to decode input image
|
139 |
def decode_base64_image(self, image_string):
|
|
|
124 |
image=control_image,
|
125 |
num_inference_steps=num_inference_steps,
|
126 |
guidance_scale=guidance_scale,
|
|
|
127 |
height=height,
|
128 |
width=width,
|
129 |
controlnet_conditioning_scale=controlnet_conditioning_scale,
|
|
|
132 |
|
133 |
# generator=self.generator COMMENTED from self.pipe
|
134 |
# return the first generated PIL image
|
135 |
+
return out.images[0]
|
136 |
|
137 |
# helper to decode input image
|
138 |
def decode_base64_image(self, image_string):
|