Image build is optimized.

flask_ml_api/Dockerfile

@@ -14,13 +14,17 @@ ENV PATH="/opt/myenv/bin:$PATH"
 WORKDIR /app
 COPY . /app
 COPY ./asyst /app/asyst
-COPY requirements.txt .
 
-# Install dependencies
-RUN /opt/myenv/bin/pip install -r /app/requirements.txt \
+# Install dependencies in separate steps to reduce memory usage
+RUN /opt/myenv/bin/pip install torch \
+    && /opt/myenv/bin/pip install Flask \
+    && /opt/myenv/bin/pip install matplotlib \
+    && /opt/myenv/bin/pip install scikit-learn \
+    && /opt/myenv/bin/pip install transformers \
+    && /opt/myenv/bin/pip install pandas \
+    && /opt/myenv/bin/pip install sentence_transformers \
     && /opt/myenv/bin/python3 -m pip install --upgrade setuptools wheel
 
-
 # Final image
 FROM python:3.10-slim