FROM python:3.11-slim

WORKDIR /app

COPY train.py /app/train.py
COPY prepare.py /app/prepare.py

CMD ["python3", "train.py"]
