Update Dockerfile

Dockerfile

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