fix(api): check image size before blending

ssube · Jan 18, 2023 · 08dbc0c · 08dbc0c
1 parent d6f2c62
commit 08dbc0c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/api/onnx_web/pipeline.py b/api/onnx_web/pipeline.py
@@ -182,7 +182,11 @@ def run_inpaint_pipeline(
         num_inference_steps=params.steps,
         width=size.width,
     ).images[0]
-    image = ImageChops.blend(source_image, image, strength)
+
+    if image.size == source_image.size:
+        image = ImageChops.blend(source_image, image, strength)
+    else:
+        print('output image size does not match source, skipping post-blend')
 
     if upscale.faces or upscale.scale > 1:
         image = upscale_resrgan(ctx, upscale, image)