karimbenharrak commited on
Commit
7c87b47
1 Parent(s): 744d735

Update handler.py

Browse files
Files changed (1) hide show
  1. 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