Dockerfile

FROM python:3.9

WORKDIR /code

COPY ./requirements.txt /code/requirements.txt

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

COPY . /code

# Create a writable directory for cache
RUN mkdir -p /code/cache && chmod 777 /code/cache
ENV TORCH_HOME=/code/cache

CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]