diff --git a/Dockerfile b/Dockerfile
index 2d8fc10..f5d6b66 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -36,8 +36,8 @@ RUN pip3 install --no-cache-dir \
     opencv-python \
     scikit-learn \
     tf-nightly \
-    tf-keras \
-    transformers
+    transformers \
+    keras==2.14.0
 
 
 # Install nltk data