diff --git a/ai-moderation/Dockerfile b/ai-moderation/Dockerfile
index 68d3bc0..96161bf 100644
--- a/ai-moderation/Dockerfile
+++ b/ai-moderation/Dockerfile
@@ -12,11 +12,12 @@ RUN pip install -r requirements.txt
 
 COPY app ./app
 
-RUN useradd -r -u 1001 app \
+RUN useradd -r -m -u 1001 app \
  && mkdir -p /models \
  && chown app /models
 USER app
 
 ENV MODEL_CACHE_DIR=/models
+ENV HF_HOME=/models/hf
 EXPOSE 8000
 CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000"]
\ No newline at end of file