FROM python:3.11-slim

WORKDIR /workspace
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt
COPY . .

CMD ["python", "run.py", "--train", "/data/train.jsonl", "--val", "/data/val.jsonl", "--weights", "/tmp/weights", "--output", "/tmp/results.json"]
