FROM python:3.10

RUN pip install torch torchvision torchaudio
RUN pip install transformers
RUN pip install numpy scipy

# Can also be set to /bin/sh if pip is needed first
ENTRYPOINT ["/usr/local/bin/python"]