FROM pytorch/pytorch:2.5.1-cuda12.1
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