apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

feat(api): add ready endpoint to check output status

This commit is contained in:
Sean Sube 2023-01-12 19:56:41 -06:00
parent 7e35b7b34f
commit 55e8b800d2
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/onnx_web/serve.py
View File

@ -447,10 +447,12 @@ def inpaint():
}) })
@app.route('/ready/<path:filename>') @app.route('/ready')
def ready(filename): def ready():
output_file = request.args.get('output', None)
return json_with_cors({ return json_with_cors({
'ready': executor.futures.done(filename), 'ready': executor.futures.done(output_file),
}) })