lsb commited on
Commit
c734fae
1 Parent(s): 80209ba

parameterize inpaint size

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -72,8 +72,8 @@ def app(img, prompt, num_inference_steps, seed, inpaint_size):
72
  #print(prompt.__class__, img.__class__, mask.__class__, img.shape, mask.shape, mask.dtype, img.dtype)
73
  overlay_img = inpainting_pipeline(
74
  prompt=prompt,
75
- image=Image.fromarray(img).resize(inpaint_size),
76
- mask_image=(mask).resize(inpaint_size),
77
  strength=1,
78
  num_inference_steps=num_inference_steps,
79
  height=inpaint_size,
 
72
  #print(prompt.__class__, img.__class__, mask.__class__, img.shape, mask.shape, mask.dtype, img.dtype)
73
  overlay_img = inpainting_pipeline(
74
  prompt=prompt,
75
+ image=Image.fromarray(img).resize((inpaint_size, inpaint_size)),
76
+ mask_image=(mask).resize((inpaint_size, inpaint_size)),
77
  strength=1,
78
  num_inference_steps=num_inference_steps,
79
  height=inpaint_size,