Dockerfile

FROM python:3.8

RUN pip install flask transformers

COPY . .

ENV MODEL_NAME bert-base-cased
ENV API_PORT 8080

CMD ["python", "app.py"]