File size: 774 Bytes
5a7000a
22a5d58
4308fb7
5a7000a
 
 
 
 
a1780ec
 
5a7000a
 
 
a1780ec
5a7000a
 
 
 
 
a20c6fa
 
 
8290e06
 
 
434858f
12a7e19
a20c6fa
 
5a7000a
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
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" ]