| 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 |
| WORKDIR $HOME/app |
|
|
|
|
| ENV HF_HOME=$HOME/.cache/huggingface |
| ENV TRANSFORMERS_CACHE=$HF_HOME |
|
|
|
|
| RUN python3 -c "from huggingface_hub import snapshot_download; import os; snapshot_download('MBZUAI/LaMini-Flan-T5-77M', cache_dir=os.environ.get('HF_HOME'))" |
|
|
|
|
| COPY --chown=user . $HOME/app |
|
|
|
|
| EXPOSE 7860 |
| CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"] |
