| FROM nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04 | |
| RUN useradd -m -u 1000 user | |
| WORKDIR /subgen | |
| ADD https://raw.githubusercontent.com/McCloudS/subgen/main/requirements.txt /subgen/requirements.txt | |
| RUN apt-get update \ | |
| && apt-get install -y \ | |
| python3 \ | |
| python3-pip \ | |
| ffmpeg \ | |
| && apt-get clean \ | |
| && rm -rf /var/lib/apt/lists/* \ | |
| && pip3 install -r requirements.txt | |
| ENV PYTHONUNBUFFERED=1 | |
| ADD https://raw.githubusercontent.com/McCloudS/subgen/main/launcher.py /subgen/launcher.py | |
| ADD https://raw.githubusercontent.com/McCloudS/subgen/main/subgen.py /subgen/subgen.py | |
| COPY --chown=user . $HOME/subgen | |
| COPY --chown=user . ./models | |
| RUN mkdir -p /models && chmod 777 /models | |
| USER user | |
| CMD [ "bash", "-c", "python3 -u launcher.py" ] | |