From 9c93e16698d2277026b76dd56068bcfd55de53cb Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sun, 15 Jan 2023 11:43:47 -0600
Subject: [PATCH] fix(api): clean up background jobs once they are ready (for
 #55)

---
 api/onnx_web/serve.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py
index 0c6e77a3..503a07cd 100644
--- a/api/onnx_web/serve.py
+++ b/api/onnx_web/serve.py
@@ -605,9 +605,13 @@ def inpaint():
 @app.route('/api/ready')
 def ready():
     output_file = request.args.get('output', None)
+    done = executor.futures.done(output_file)
+
+    if done == True:
+        executor.futures.pop(output_file)
 
     return jsonify({
-        'ready': executor.futures.done(output_file),
+        'ready': done,
     })