Dockerfile

FROM python:3.10.11

WORKDIR /app

RUN apt-get update && apt-get install -y \
    build-essential \
    curl \
    software-properties-common \
    git \
    && rm -rf /var/lib/apt/lists/*

COPY requirements.txt ./
COPY src/ ./src/

RUN pip3 install -r requirements.txt
RUN pip3 install torch==2.7.0 --index-url https://download.pytorch.org/whl/cu126
RUN pip3 install torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/cu126
RUN pip3 install torchvision==0.22.0 --index-url https://download.pytorch.org/whl/cu126

EXPOSE 8501

HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health

ENTRYPOINT ["streamlit", "run", "src/app.py", "--server.port=8501", "--server.address=0.0.0.0"]