From 4911a6cbd42c4ff6fcabd4d2a7c1c84de7772c3f Mon Sep 17 00:00:00 2001
From: Artem Baranovskyi <artem.baranovsky1980@gmail.com>
Date: Sat, 11 Jan 2025 19:51:27 +0200
Subject: [PATCH] Image build is optimized.

---
 .gitlab-ci.yml          | 3 ++-
 flask_ml_api/Dockerfile | 4 +---
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index b1551ba..a40fa5f 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -41,7 +41,8 @@ build:
     - docker system prune -a -f
     - docker volume prune -f
   script:
-    - docker build --cache-from=asyst-nlp-grader --no-cache --progress=plain -t asyst-nlp-grader -f flask_ml_api/Dockerfile .
+#    - docker build --cache-from=asyst-nlp-grader --no-cache --progress=plain -t asyst-nlp-grader -f flask_ml_api/Dockerfile .
+    - docker buildx build --cache-from=type=registry,ref=22baar1mst/asyst-nlp-grader --no-cache --progress=plain -t 22baar1mst/asyst-nlp-grader:latest -f flask_ml_api/Dockerfile --push .
     - docker push 22baar1mst/asyst-nlp-grader:latest
   cache:
     key: build-cache
diff --git a/flask_ml_api/Dockerfile b/flask_ml_api/Dockerfile
index c1ca870..d6fa706 100755
--- a/flask_ml_api/Dockerfile
+++ b/flask_ml_api/Dockerfile
@@ -11,12 +11,10 @@ ENV PATH="/opt/myenv/bin:$PATH"
 WORKDIR /app
 COPY . /app
 
-# Install dependencies in smaller steps
 RUN pip install --no-cache-dir Flask matplotlib scikit-learn transformers pandas sentence_transformers
 RUN pip install --no-cache-dir --upgrade setuptools wheel
 
-# Final image
-FROM python:3.10-slim
+FROM python:3.10-slim AS final
 
 COPY --from=builder /opt/myenv /opt/myenv
 COPY --from=builder /app /app
-- 
GitLab