Dockerfile

FROM python:3.11

COPY requirements.txt requirements.txt
RUN apt-get update && apt-get install -y --no-install-recommends \
      bzip2 \
      g++ \
      git \
      graphviz \
      libgl1-mesa-glx \
      libhdf5-dev \
      openmpi-bin \
      wget \
      python3-tk && \
    rm -rf /var/lib/apt/lists/*

RUN pip install --upgrade pip

RUN pip install torch==2.1.0

RUN pip install --no-cache-dir -r requirements.txt

RUN useradd -m -u 1000 myuser


USER myuser

COPY --chown=myuser app app

EXPOSE 8001

ENTRYPOINT ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8001"]