Dockerfile

FROM python:3.9-slim

WORKDIR /app

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

# Copy files
COPY requirements.txt ./
COPY resnet18_cat.pth ./
COPY src/ ./src/

# Install Python dependencies
RUN pip3 install -r requirements.txt

# Expose Streamlit default port
EXPOSE 8501

# Healthcheck (optional)
HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health

# Run the Streamlit app using app.py
ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]