Dockerfile

FROM python:3.9

WORKDIR /app

RUN mkdir /app/.cache
RUN chmod -R 777 /app/.cache
ENV HF_HOME=/app/.cache

COPY ./requirements.txt /app
RUN pip install -r requirements.txt

COPY ./checkpoint-600/ /checkpoint-600/
COPY ./app.py /app

EXPOSE 7860
CMD ["flask", "run" ,"--host", "0.0.0.0", "--port", "7860"]