diff --git a/api/onnx_web/chain/pipeline.py b/api/onnx_web/chain/pipeline.py index 1a6a597b..c785fce0 100644 --- a/api/onnx_web/chain/pipeline.py +++ b/api/onnx_web/chain/pipeline.py @@ -235,8 +235,9 @@ class ChainPipeline: **kwargs, ) + metadata = stage_sources.metadata stage_sources = StageResult( - images=stage_results, metadata=stage_sources.metadata + images=stage_results, metadata=metadata ) else: logger.debug( diff --git a/api/onnx_web/server/api.py b/api/onnx_web/server/api.py index 1166c0dd..26777ce6 100644 --- a/api/onnx_web/server/api.py +++ b/api/onnx_web/server/api.py @@ -129,7 +129,7 @@ def image_reply( } if outputs is not None: - if metadata is not None: + if metadata is None: logger.error("metadata is required with outputs") return error_reply("metadata is required with outputs")