Dockerfile

FROM python:3.9

# Create app directory
WORKDIR /code

RUN wget https://huggingface.co/spaces/anon5/vit-base-patch32-384-api/raw/main/app.py -O ./app.py
RUN pip install flask transformers

EXPOSE 5000
CMD [ "python", "-m", "flask ", "--app", "app", "run" ]