apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): always apply mask filter for inpainting

This commit is contained in:
Sean Sube 2023-01-15 11:39:40 -06:00
parent b1aca928ab
commit e4020cf3f6
1 changed files with 11 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
api/onnx_web/serve.py
View File

@ -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,