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 b89eb42712
commit 41935667c4
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>')
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),
})