Update Dockerfile

Dockerfile

@@ -1,11 +1,31 @@
-FROM python:3.10-slim
+FROM python:3.11
 
-RUN apt update && apt install -y git && rm -rf /var/lib/apt/lists/*
-WORKDIR /app
-COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
 
-COPY . .
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user PATH=$HOME/.local/bin:$PATH
+
+
+RUN mkdir -p $HOME/.cache/huggingface
+ENV HF_HOME=$HOME/.cache/huggingface
+WORKDIR $HOME/app
+
+
+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
+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)
+EOF
+
+
+COPY --chown=user . .
 
 EXPOSE 7860
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]