| FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04 | |
| ENV DEBIAN_FRONTEND=noninteractive | |
| ENV TZ=Europe/Paris | |
| RUN apt-get update | |
| RUN apt-get install -y git git-lfs wget curl make cmake ffmpeg llvm python3 python3-pip | |
| RUN apt-get install -y build-essential xz-utils tk-dev libsm6 libxext6 libgl1-mesa-glx | |
| RUN apt-get install -y libssl-dev zlib1g-dev libbz2-dev libreadline-dev libsqlite3-dev libncursesw5-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev | |
| RUN rm -rf /var/lib/apt/lists/* | |
| RUN git lfs install | |
| RUN useradd -m -u 1000 user | |
| USER user | |
| ENV HOME=/home/user | |
| ENV PATH=/home/user/.local/bin:$PATH | |
| WORKDIR /home/user/app | |
| COPY . /home/user/app | |
| RUN pip install --no-cache-dir addict albumentations addict albumentations basicsr blendmodes einops gradio imageio imageio-ffmpeg invisible-watermark kornia numba omegaconf open_clip_torch prettytable pytorch-lightning safetensors streamlit streamlit-drawable-canvas test-tube timm torchmetrics transformers torch torchvision webdataset xformers yapf | |
| ENV PYTHONPATH=$HOME/app | |
| ENV PYTHONUNBUFFERED=1 | |
| ENV GRADIO_ALLOW_FLAGGING=never | |
| ENV GRADIO_NUM_PORTS=1 | |
| ENV GRADIO_SERVER_NAME=0.0.0.0 | |
| ENV GRADIO_THEME=huggingface | |
| ENV SYSTEM=spaces | |
| RUN python3 app.py |