diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py index a2934f6b..b91a0a8e 100644 --- a/api/onnx_web/serve.py +++ b/api/onnx_web/serve.py @@ -447,10 +447,12 @@ def inpaint(): }) -@app.route('/ready/') -def ready(filename): +@app.route('/ready') +def ready(): + output_file = request.args.get('output', None) + return json_with_cors({ - 'ready': executor.futures.done(filename), + 'ready': executor.futures.done(output_file), })