diff --git a/MLBaseDockerfile b/MLBaseDockerfile
index d1380f9..fbd9f63 100644
--- a/MLBaseDockerfile
+++ b/MLBaseDockerfile
@@ -1,11 +1,11 @@
 FROM condaforge/miniforge3:25.9.1-0
 
 WORKDIR /app
-RUN pip install huggingface_hub
+RUN pip install --root-user-action ignore 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 hf download sentence-transformers/all-distilroberta-v1 --local-dir /assets/models/sentence-transformers/all-distilroberta-v1
+RUN hf download cross-encoder/qnli-distilroberta-base --local-dir /assets/models/cross-encoder/qnli-distilroberta-base
 
 RUN conda install -c conda-forge \
     numpy=1.26.4 \