apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): keep chain pipeline images and use request size

This commit is contained in:
Sean Sube 2023-02-05 10:27:34 -06:00
parent bf1378bf96
commit 9d0609fefe
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/serve.py
View File

@ -681,7 +681,7 @@ def chain():
)
source_file = request.files.get(stage_source_name)
source_image = Image.open(BytesIO(source_file.read())).convert("RGB")
source_image = source_image.thumbnail((512, 512))
source_image.thumbnail((size.width, size.height))
kwargs["source_image"] = source_image
if stage_mask_name in request.files:
@ -692,7 +692,7 @@ def chain():
)
mask_file = request.files.get(stage_mask_name)
mask_image = Image.open(BytesIO(mask_file.read())).convert("RGB")
mask_image = mask_image.thumbnail((512, 512))
mask_image.thumbnail((size.width, size.height))
kwargs["mask_image"] = mask_image
pipeline.append((callback, stage, kwargs))