diff --git a/requirements.txt b/requirements.txt
index 4a84621..1c60281 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,4 +6,5 @@ matplotlib
 seaborn
 sacred
 pymongo
-mlflow
\ No newline at end of file
+mlflow
+torchvision
\ No newline at end of file