Update Dockerfile

Dockerfile

@@ -1,25 +1,39 @@
-FROM python:3.11
+FROM python:3.11-slim
 
 
 RUN useradd -m -u 1000 user
+
+
+WORKDIR /tmp_build
+COPY requirements.txt .
+
+RUN pip install --no-cache-dir --upgrade pip \
+ && pip install --no-cache-dir -r requirements.txt
+
+
+RUN mkdir -p /home/user/.cache/huggingface \
+ && chown -R user:user /home/user/.cache
+
+
 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
 
+ENV TRANSFORMERS_CACHE=$HF_HOME
 
-COPY --chown=user requirements.txt .
-RUN pip install --no-cache-dir --upgrade pip \
- && pip install --no-cache-dir -r requirements.txt
 
+RUN python -c "from huggingface_hub import snapshot_download; import os; snapshot_download('MBZUAI/LaMini-Flan-T5-77M', cache_dir=os.environ.get('HF_HOME'))"
+
+cache_dir = os.environ.get("HF_HOME")
+snapshot_download(repo_id="MBZUAI/LaMini-Flan-T5-77M", cache_dir=cache_dir)
+EOF
 
-RUN python3 -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='MBZUAI/LaMini-Flan-T5-77M')"
 
+COPY --chown=user . $HOME/app
 
-COPY --chown=user . .
 
 EXPOSE 7860
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]