Dockerfile

# syntax=docker/dockerfile:1

FROM tensorflow/tensorflow:latest

RUN useradd -ms /bin/bash admin

ENV TRANSFORMERS_CACHE=/code/.cache
WORKDIR /code

USER 0
RUN mkdir /code/.cache
RUN chown -R admin:admin /code/.cache
RUN chmod 755 /code/.cache
USER admin

COPY requirements.txt requirements.txt
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt

COPY . .

CMD [ "python3", "-m" , "flask", "run", "--host=0.0.0.0","--port", "7860"]