pin keras to 2.14.0

cs50 · Mar 4, 2024 · 3ec6eab · 3ec6eab
1 parent 3b2be23
commit 3ec6eab
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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