diff --git a/Dockerfile b/Dockerfile
index 2fa6868..2d8fc10 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -36,6 +36,7 @@ RUN pip3 install --no-cache-dir \
     opencv-python \
     scikit-learn \
     tf-nightly \
+    tf-keras \
     transformers