diff --git a/Dockerfile b/Dockerfile
index e48d2a9..18baadd 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -9,7 +9,8 @@ RUN apt-get update \
     && pip3 install --user kaggle \
     && pip3 install --user pandas \
     && pip3 install --user seaborn \
-    && pip3 install --user scikit-learn
+    && pip3 install --user scikit-learn \
+    && pip3 install --user tensorflow
 
 ENV PATH="/root/.local/bin:$PATH"