diff --git a/expo/data/dataset.py b/expo/data/dataset.py
index 8ad6f2854..f2f01f71b 100644
--- a/expo/data/dataset.py
+++ b/expo/data/dataset.py
@@ -26,7 +26,7 @@ TEXT_MODALITY = """
 """
 
 IMAGE_MODALITY = """
-7. You could use models from torchvision library for this image dataset.
+7. You could use models from transformers library for this image dataset.
 8. Use gpu if available for faster training.
 """