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