diff --git a/Dockerfile b/Dockerfile
index e0b5d2a..64ca3d9 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -18,6 +18,7 @@ RUN pip3 install  wget
 RUN pip3 install  keras
 RUN pip3 install  GitPython
 RUN pip3 install  pymongo
+RUN pip3 install  mlflow
 
 WORKDIR /app