Skip to content

Commit

Permalink
fix(api): always apply mask filter for inpainting
Browse files Browse the repository at this point in the history
  • Loading branch information
ssube committed Jan 15, 2023
1 parent b1aca92 commit e4020cf
Showing 1 changed file with 12 additions and 13 deletions.
25 changes: 12 additions & 13 deletions api/onnx_web/serve.py
Original file line number Diff line number Diff line change
Expand Up @@ -313,19 +313,18 @@ def run_inpaint_pipeline(
latents = get_latents_from_seed(seed, width, height)
rng = np.random.RandomState(seed)

if left > 0 or right > 0 or top > 0 or bottom > 0:
print('expanding image for outpainting')
source_image, mask_image, noise_image, _full_dims = expand_image(
source_image,
mask_image,
(left, right, top, bottom),
noise_source=noise_source,
mask_filter=mask_filter)

if environ.get('DEBUG') is not None:
source_image.save('./last-source.png')
mask_image.save('./last-mask.png')
noise_image.save('./last-noise.png')
print('applying mask filter and generating noise source')
source_image, mask_image, noise_image, _full_dims = expand_image(
source_image,
mask_image,
(left, right, top, bottom),
noise_source=noise_source,
mask_filter=mask_filter)

if environ.get('DEBUG') is not None:
source_image.save('./last-source.png')
mask_image.save('./last-mask.png')
noise_image.save('./last-noise.png')

image = pipe(
prompt,
Expand Down

0 comments on commit e4020cf

Please sign in to comment.