From e4020cf3f6d0817ba04eb30f17c7cf1b587d31b7 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 15 Jan 2023 11:39:40 -0600 Subject: [PATCH] fix(api): always apply mask filter for inpainting --- api/onnx_web/serve.py | 23 +++++++++++------------ 1 file changed, 11 insertions(+), 12 deletions(-) diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py index aca3e6a6..0c6e77a3 100644 --- a/api/onnx_web/serve.py +++ b/api/onnx_web/serve.py @@ -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) + 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') + 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,