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"]