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

ssube · Jan 13, 2023 · 55e8b80 · 55e8b80
1 parent 7e35b7b
commit 55e8b80
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py
@@ -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),
     })