Image build is optimized.

flask_ml_api/Dockerfile

@@ -12,15 +12,18 @@ ENV PATH="/opt/myenv/bin:$PATH"
 WORKDIR /app
 COPY . /app
 
-RUN pip install --no-cache-dir Flask && \
-    pip install --no-cache-dir matplotlib && \
-    pip install --no-cache-dir scikit-learn && \
-    pip install --no-cache-dir transformers && \
-    pip install --no-cache-dir pandas && \
-    pip install --no-cache-dir sentence_transformers && \
-    pip install --no-cache-dir --upgrade setuptools wheel
-
-RUN rm -rf /root/.cache/pip
+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 && \
+#    pip install --no-cache-dir transformers && \
+#    pip install --no-cache-dir pandas && \
+#    pip install --no-cache-dir sentence_transformers && \
+#    pip install --no-cache-dir --upgrade setuptools wheel
+#RUN rm -rf /root/.cache/pip
 
 
 FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime