diff --git a/src/main.py b/src/main.py
index 29f0e84..8716918 100644
--- a/src/main.py
+++ b/src/main.py
@@ -43,6 +43,9 @@ def post_detect(file: Image, strength: DetectionStrength):
     print("\nWorking on: " + file_name)
     print("\nStrength: " + strength + "\n")
 
+    if not os.path.exists("data/"):
+        os.makedirs("data")
+
     with open("data/" + file_name, "wb") as f:
         f.write(b64decode(file.file_contents))