Dockerfile

FROM nvcr.io/nvidia/pytorch:24.10-py3

# Install tools
RUN apt-get update && apt-get install -y git ffmpeg wget build-essential ninja-build && \
    apt-get clean && rm -rf /var/lib/apt/lists/*

COPY requirements.txt /requirements.txt
COPY app.py /app/app.py
COPY setup_runtime.sh /setup_runtime.sh
RUN chmod +x /setup_runtime.sh

CMD ["/bin/bash", "/setup_runtime.sh"]