Dockerfile

FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu22.04

ENV DEBIAN_FRONTEND=noninteractive     PYTHONUNBUFFERED=1     PIP_NO_CACHE_DIR=1     HOME=/home/user     PATH=/home/user/.local/bin:$PATH     FORCE_CPU=0

RUN apt-get update && apt-get install -y     python3     python3-pip     ffmpeg     libgl1     libglib2.0-0     && rm -rf /var/lib/apt/lists/*

RUN useradd -m -u 1000 user
USER user
WORKDIR /home/user/app

COPY --chown=user requirements.txt ./requirements.txt
RUN python3 -m pip install --upgrade pip && python3 -m pip install -r requirements.txt

COPY --chown=user . /home/user/app

EXPOSE 7860
CMD ["python3", "app.py"]