diff --git a/api/onnx_web/chain/base.py b/api/onnx_web/chain/base.py index b939f028..9c0cb76e 100644 --- a/api/onnx_web/chain/base.py +++ b/api/onnx_web/chain/base.py @@ -105,7 +105,9 @@ class ChainPipeline: """ DEPRECATED: use `run` instead """ - if callback is not None: + if callback is None: + callback = worker.get_progress_callback() + else: callback = ChainProgress.from_progress(callback) start = monotonic()