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