diff --git a/Dockerfile b/Dockerfile
index 47596ba..d63e477 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,7 +2,7 @@ FROM ubuntu:latest
 
 RUN apt-get update && \
     apt-get install -y python3-pip && \
-    pip3 install kaggle pandas scikit-learn tensorflow matplotlib
+    pip3 install kaggle pandas scikit-learn tensorflow matplotlib mlflow
 
 RUN useradd -ms /bin/bash jenkins