diff --git a/Dockerfile b/Dockerfile
index 2baa099..d33be64 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -13,7 +13,7 @@ RUN pip install --upgrade pip
 
 
 
-RUN pip install torch pandas numpy scikit-learn kaggle
+RUN pip install torch pandas numpy scikit-learn kaggle mlflow
 
 
 COPY . /