diff --git a/Dockerfile b/Dockerfile
index 106015f..0c2b6c9 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,4 +2,4 @@ FROM ubuntu:20.04
 
 RUN apt update
 RUN apt install -y python3 python3-pip
-RUN pip3 install mlflow 
+RUN pip3 install mlflow sklearn