diff --git a/requirements.txt b/requirements.txt
index 956f3a0..a924067 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,3 +4,5 @@ Pillow
 towhee
 timm
 transformers>=4.26.0
+datasets
+datasets[vision]