FROM --platform=linux/arm64 python:3.10-slim

WORKDIR /

RUN pip install --no-cache-dir --upgrade pip \
 && pip install --no-cache-dir torch torchvision \
 && pip install --no-cache-dir tqdm

RUN mkdir -p /src /data /output
CMD ["bash"]