Update handler.py
Browse files- handler.py +4 -1
handler.py
CHANGED
@@ -165,12 +165,15 @@ class EndpointHandler():
|
|
165 |
|
166 |
# generate image
|
167 |
image = pipe(
|
168 |
-
prompt,
|
|
|
169 |
num_inference_steps=num_inference_steps,
|
170 |
eta=1.0,
|
171 |
image=image,
|
172 |
mask_image=mask_image,
|
173 |
control_image=control_image,
|
|
|
|
|
174 |
).images[0]
|
175 |
|
176 |
return image
|
|
|
165 |
|
166 |
# generate image
|
167 |
image = pipe(
|
168 |
+
prompt=prompt,
|
169 |
+
negative_prompt=negative_prompt,
|
170 |
num_inference_steps=num_inference_steps,
|
171 |
eta=1.0,
|
172 |
image=image,
|
173 |
mask_image=mask_image,
|
174 |
control_image=control_image,
|
175 |
+
guidance_scale,
|
176 |
+
strength=strength
|
177 |
).images[0]
|
178 |
|
179 |
return image
|