Update Dockerfile

Dockerfile

@@ -1,31 +1,36 @@
-FROM python:3.11
+FROM python:3.11-slim
 
 
 RUN useradd -m -u 1000 user
+
+
 USER user
-ENV HOME=/home/user PATH=$HOME/.local/bin:$PATH
+ENV HOME=/home/user
+WORKDIR $HOME/app
 
 
-RUN mkdir -p $HOME/.cache/huggingface
 ENV HF_HOME=$HOME/.cache/huggingface
-WORKDIR $HOME/app
+ENV TRANSFORMERS_CACHE=$HF_HOME
+RUN mkdir -p $HF_HOME
+
+COPY --chown=user requirements.txt $HOME/app/requirements.txt
 
 
-COPY --chown=user requirements.txt .
 RUN pip install --no-cache-dir --upgrade pip \
  && pip install --no-cache-dir -r requirements.txt
 
 
-RUN pip install --no-cache-dir huggingface_hub \
- && python3 - << 'EOF'
-import os
+RUN python - << 'EOF'
 from huggingface_hub import snapshot_download
-cache_dir = os.environ.get("HF_HOME", os.path.expanduser("~/.cache/huggingface"))
-snapshot_download("rasyosef/Phi-1_5-Instruct-v0.1", cache_dir=cache_dir)
+
+snapshot_download("rasyosef/Phi-1_5-Instruct-v0.1")
 EOF
 
 
-COPY --chown=user . .
+COPY --chown=user . $HOME/app
+
 
 EXPOSE 7860
+
+
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]