diff --git a/sora.py b/sora.py
index 49ef924..fa4679e 100644
--- a/sora.py
+++ b/sora.py
@@ -4,6 +4,7 @@ import torch
 import numpy as np
 import av
 from PIL import Image
+import os
 from openai import OpenAI
 
 class SoraGen: