ViT-Explainer / Dockerfile
JuanHernandez-uc
Add application file
6c80045
FROM python:3.11-slim
ENV PYTHONDONTWRITEBYTECODE=1 \
PYTHONUNBUFFERED=1 \
PIP_NO_CACHE_DIR=1 \
PORT=7860 \
HF_HOME=/home/user/.cache/huggingface \
TORCH_HOME=/home/user/.cache/torch \
OMP_NUM_THREADS=1 \
MKL_NUM_THREADS=1
RUN apt-get update && apt-get install -y --no-install-recommends \
ca-certificates \
libgomp1 \
&& rm -rf /var/lib/apt/lists/*
RUN useradd -m -u 1000 user
USER user
ENV HOME=/home/user \
PATH=/home/user/.local/bin:$PATH
WORKDIR $HOME/app
COPY --chown=user:user requirements.txt .
RUN pip install --upgrade pip && pip install -r requirements.txt
COPY --chown=user:user . .
EXPOSE 7860
CMD ["sh", "-c", "uvicorn app:app --host 0.0.0.0 --port ${PORT:-7860} --workers 1"]