diff --git a/api/onnx_web/output.py b/api/onnx_web/output.py index 7faf301e..024b29f3 100644 --- a/api/onnx_web/output.py +++ b/api/onnx_web/output.py @@ -69,7 +69,7 @@ def save_result( base_name: str, ) -> List[str]: results = [] - for i, image, metadata in enumerate(zip(result.as_image(), result.metadata)): + for i, (image, metadata) in enumerate(zip(result.as_image(), result.metadata)): results.append( save_image( server, diff --git a/api/onnx_web/server/api.py b/api/onnx_web/server/api.py index 2e431ff8..670e675d 100644 --- a/api/onnx_web/server/api.py +++ b/api/onnx_web/server/api.py @@ -112,13 +112,13 @@ def image_reply( metadata: List[ImageMetadata] = None, ): if stages is None: - stages = Progress() + stages = Progress(0, 0) if steps is None: - steps = Progress() + steps = Progress(0, 0) if tiles is None: - tiles = Progress() + tiles = Progress(0, 0) data = { "name": name, @@ -688,7 +688,9 @@ def register_api_routes(app: Flask, server: ServerContext, pool: DevicePoolExecu return [ app.route("/api")(wrap_route(introspect, server, app=app)), # job routes - app.route("/api/job", methods=["POST"])(wrap_route(job_create, server, pool=pool)), + app.route("/api/job", methods=["POST"])( + wrap_route(job_create, server, pool=pool) + ), app.route("/api/job/cancel", methods=["PUT"])( wrap_route(job_cancel, server, pool=pool) ),