Dockerfile

FROM python:3.10

WORKDIR /app

ENV NUMBA_CACHE_DIR=/app/cache
RUN mkdir -p /app/cache
RUN chmod -R 777 /app/*

COPY ./requirements.txt requirements.txt

RUN pip install --no-cache-dir torch torchvision torchaudio
RUN pip install --no-cache-dir -r requirements.txt

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