added tensorflow and other ML packages

cs50 · Dec 30, 2023 · e73fe91 · e73fe91
1 parent ac0f04b
commit e73fe91
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -5,7 +5,7 @@ USER root
 ARG DEBIANFRONTEND=noninteractive
 
 
-RUN apt-get update -qq && apt-get install -y jq
+RUN apt-get update -qq && apt-get install -y jq cmake
 
 
 # Suggested build environment for Python, per pyenv, even though we're building ourselves
@@ -60,14 +60,12 @@ RUN apt update && \
 # TODO remove werkzeug after https://github.com/fengsp/flask-session/issues/99 is fixed
 RUN pip3 install --no-cache-dir \
         flask_sqlalchemy \
-        nltk \
         numpy \
         pandas \
         passlib \
         plotly \
         pytz \
         cffi \
-        opencv-python \
         inflect \
         emoji \
         pyfiglet \
@@ -79,6 +77,16 @@ RUN pip3 install --no-cache-dir \
         fpdf2==2.7.6
 
 
+# Install ML packages for CS50 AI
+RUN pip3 install --no-cache-dir \
+    nltk \
+    opencv-python \
+    scikit-learn \
+    tf-nightly \
+    transformers && \
+    python3 -m nltk.downloader -d /usr/share/nltk_data/ punkt
+
+
 # Install CS50 Python packages
 RUN pip3 install cs50 --upgrade --no-cache-dir