Updated coffeehouse_nsfw to Delete unused uploads

2021-01-16 15:45:01 -05:00 · 2021-01-16 15:45:01 -05:00 · 8ccd8ef874
parent 1a8f1a0343
commit 8ccd8ef874
1 changed files with 1 additions and 0 deletions
--- a/services/nsfw_detection/coffeehouse_nsfw/server.py
+++ b/services/nsfw_detection/coffeehouse_nsfw/server.py
@ -43,6 +43,7 @@ class Server(object):
            f.write(base64.b64decode(post_data["input"]))

        results = self.classifier.classify(tmp_image)[tmp_image]
+        os.remove(tmp_image)

        response = {
            "status": True,