diff --git a/requirements.txt b/requirements.txt
index 91f8f77..f81813d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,3 +3,5 @@ torch
 towhee
 towhee.models
 transformers
+datasets
+datasets[vision]