src/train_docker/Dockerfile

# Use official Python 3.12.11 image as base
FROM python:3.12.11-slim

# Set working directory inside the container
WORKDIR /src/train_docker

# Copy requirements.txt first (for better caching)
COPY requirements.txt .

# Install dependencies
RUN pip install --upgrade pip && \
    pip install -r requirements.txt

# Copy the rest of the application code
COPY . .

# Default command to run your training script
CMD ["python", "train.py"]