diff --git a/requirements.txt b/requirements.txt
index 4849fd9..0e6dc23 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,3 +8,4 @@ gradio==3.9
 deepspeed==0.9.2 
 accelerate
 tensorboardX
+datasets