| FROM python:3.11-slim |
|
|
|
|
| RUN useradd -m -u 1000 user |
|
|
|
|
| ENV HOME=/home/user |
| 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 . |
| 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'))" |
|
|
|
|
| USER user |
|
|
|
|
| WORKDIR $HOME/app |
| COPY --chown=user . $HOME/app |
|
|
|
|
| EXPOSE 7860 |
|
|
|
|
| CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"] |