diff --git a/flask_ml_api/Dockerfile b/flask_ml_api/Dockerfile
index 49dd2be619a905bbaaa8c04883e9e478095245ba..e69265fae46ac6ffea56577a965d0fc1810392ad 100755
--- a/flask_ml_api/Dockerfile
+++ b/flask_ml_api/Dockerfile
@@ -8,10 +8,6 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 RUN python3 -m venv /opt/myenv
 ENV PATH="/opt/myenv/bin:$PATH"
 
-#RUN pip install --no-cache-dir Flask matplotlib scikit-learn transformers pandas sentence_transformers && \
-#    pip install --no-cache-dir --upgrade setuptools wheel && \
-#    rm -rf /root/.cache/pip
-
 RUN pip install --no-cache-dir Flask && \
     pip install --no-cache-dir matplotlib && \
     pip install --no-cache-dir scikit-learn && \
@@ -20,6 +16,8 @@ RUN pip install --no-cache-dir Flask && \
     pip install --no-cache-dir sentence_transformers && \
     pip install --no-cache-dir --upgrade setuptools wheel
 
+RUN docker system prune -a --volumes
+
 FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
 COPY --from=build /opt/myenv /opt/myenv