From 41ec25afc1867eaaa9f6d0cc9b0c37fdd183047b Mon Sep 17 00:00:00 2001
From: Artsiom Siamashka <artsiom.siamashka@pivotics.com>
Date: Tue, 18 Nov 2025 15:40:26 +0100
Subject: [PATCH] Optimisied base image

---
 MLBaseDockerfile | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/MLBaseDockerfile b/MLBaseDockerfile
index 149cd8d..d1380f9 100644
--- a/MLBaseDockerfile
+++ b/MLBaseDockerfile
@@ -1,14 +1,15 @@
 FROM condaforge/miniforge3:25.9.1-0
 
 WORKDIR /app
-RUN mkdir /app/models
+RUN pip install huggingface_hub
+
+# download cross-encoder and sentence-transformer models to cache
+RUN hf download sentence-transformers/all-distilroberta-v1 --local-dir /assets/models/all-distilroberta-v1
+RUN hf download cross-encoder/qnli-distilroberta-base --local-dir /assets/models/qnli-distilroberta-base
+
 RUN conda install -c conda-forge \
     numpy=1.26.4 \
     pandas=2.3.3 \
     sentence-transformers=3.0.1 \
     keybert=0.9.0 \
     faiss-gpu=1.9.0
-
-# download cross-encoder and sentence-transformer models to cache
-RUN hf download sentence-transformers/all-distilroberta-v1 --local-dir /assets/models/all-distilroberta-v1
-RUN hf download cross-encoder/qnli-distilroberta-base --local-dir /assets/models/qnli-distilroberta-base