Dockerfile

FROM python:3.13.5-slim

WORKDIR /app

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

COPY requirements.txt ./
COPY src/ ./src/
COPY resnet18_cifar10_finetuned.pth ./

RUN pip3 install -r requirements.txt

# Change EXPOSE to 7860
EXPOSE 7860

# Update HEALTHCHECK to use 7860
HEALTHCHECK CMD curl --fail http://localhost:7860/_stcore/health

# Add the XSRF and CORS disable flags to ENTRYPOINT
ENTRYPOINT ["streamlit", "run", "src/image_classifier_app.py", "--server.port=7860", "--server.address=0.0.0.0", "--server.enableCORS=false", "--server.enableXsrfProtection=false"]