diff --git a/Dockerfile b/Dockerfile
index d2e849d..50241b2 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -12,9 +12,6 @@ COPY ./predictions_tf.json ./
 RUN python3 -m venv venv
 RUN /bin/bash -c "source venv/bin/activate"
 
-RUN pip install --upgrade pip
-RUN pip install tensorflow
-
-RUN apt-get install -y python3-sklearn
+RUN apt-get install -y python3-pandas python3-scikit-learn python3-tensorflow
 
 CMD ["python3", "data_processing.sh"]