apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix chain pipeline image field names

This commit is contained in:
Sean Sube 2023-02-04 14:24:18 -06:00
parent bef50d89b1
commit 2f1f6b3a11
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
api/onnx_web/serve.py
View File

@ -569,9 +569,13 @@ def upscale():
@app.route('/api/chain', methods=['POST']) @app.route('/api/chain', methods=['POST'])
def chain(): def chain():
body = request.form.get('chain') logger.debug('chain pipeline request: %s, %s',
data = yaml.safe_load(body) request.form.keys(), request.files.keys())
body = request.form.get('chain') or request.files.get('chain')
if body is None:
return error_reply('chain pipeline must have a body')
data = yaml.safe_load(body)
with open('./schema.yaml', 'r') as f: with open('./schema.yaml', 'r') as f:
schema = yaml.safe_load(f.read()) schema = yaml.safe_load(f.read())
@ -612,7 +616,7 @@ def chain():
if stage_source_name in request.files: if stage_source_name in request.files:
logger.debug('loading source image %s for pipeline stage %s', logger.debug('loading source image %s for pipeline stage %s',
stage_source_name, stage.name) stage_source_name, stage.name)
source_file = request.files.get('source') source_file = request.files.get(stage_source_name)
source_image = Image.open( source_image = Image.open(
BytesIO(source_file.read())).convert('RGB') BytesIO(source_file.read())).convert('RGB')
source_image = source_image.thumbnail((512, 512)) source_image = source_image.thumbnail((512, 512))
@ -621,7 +625,7 @@ def chain():
if stage_mask_name in request.files: if stage_mask_name in request.files:
logger.debug('loading mask image %s for pipeline stage %s', logger.debug('loading mask image %s for pipeline stage %s',
stage_mask_name, stage.name) stage_mask_name, stage.name)
mask_file = request.files.get('source') mask_file = request.files.get(stage_mask_name)
mask_image = Image.open(BytesIO(mask_file.read())).convert('RGB') mask_image = Image.open(BytesIO(mask_file.read())).convert('RGB')
mask_image = mask_image.thumbnail((512, 512)) mask_image = mask_image.thumbnail((512, 512))
kwargs['mask_image'] = mask_image kwargs['mask_image'] = mask_image