Dockerfile

# Dockerfile Public T4

FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-devel
ENV DEBIAN_FRONTEND noninteractive

WORKDIR /content
RUN pip install numexpr einops transformers k_diffusion safetensors gradio diffusers xformers

ADD . .
RUN adduser --disabled-password --gecos '' user
RUN chown -R user:user /content
RUN chmod -R 777 /content
USER user

EXPOSE 7860
CMD python /content/app.py