diff --git a/requirements.txt b/requirements.txt
index 972b430..ad32d32 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,6 +6,7 @@ accelerate
 huggingface_hub>=0.26.0
 pillow>=9.5.0
 numpy>=1.24.0
+datasets
 
 # Video processing
 opencv-python>=4.8.0