mjohanes commited on
Commit
2d7d767
·
1 Parent(s): c3c3dd4
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -186,7 +186,9 @@ else:
186
  inpainted_img = sd_pipe(
187
  prompt=prompt,
188
  image=img,
189
- mask_image=combined_mask_img
 
 
190
  ).images[0]
191
 
192
  # Update the session image to the newly inpainted result
 
186
  inpainted_img = sd_pipe(
187
  prompt=prompt,
188
  image=img,
189
+ mask_image=combined_mask_img,
190
+ width=img.width,
191
+ height=img.heigh
192
  ).images[0]
193
 
194
  # Update the session image to the newly inpainted result