apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): use requested size for initial chain pipeline input

This commit is contained in:
Sean Sube 2023-02-02 08:59:08 -06:00
parent e533dad4ba
commit 0050cea694
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

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