diff --git a/capture.py b/capture.py
index c94dfa3..bc9845c 100644
--- a/capture.py
+++ b/capture.py
@@ -2,10 +2,15 @@ import cv2
 import time
 from PIL import Image
 import numpy as np
+import os
 
 # Folder
 folder = "frames"
 
+# Create the frames folder if it doesn't exist
+frames_dir = os.path.join(os.getcwd(), folder)
+os.makedirs(frames_dir, exist_ok=True)
+
 # Initialize the webcam
 cap = cv2.VideoCapture(0)