From 0050cea6947d0753ef1f7d8ea9c57ed10daa450a Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Thu, 2 Feb 2023 08:59:08 -0600 Subject: [PATCH] fix(api): use requested size for initial chain pipeline input --- api/onnx_web/serve.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py index 240f3786..04195bbd 100644 --- a/api/onnx_web/serve.py +++ b/api/onnx_web/serve.py @@ -599,9 +599,9 @@ def chain(): logger.info('running chain pipeline with %s stages', len(pipeline.stages)) # build and run chain pipeline - fake_source = Image.new('RGB', (1, 1)) + empty_source = Image.new('RGB', (size.width, size.height)) executor.submit_stored(output, pipeline, context, - params, fake_source, output=output, size=size) + params, empty_source, output=output, size=size) return jsonify(json_params(output, params, size))