|
|
|
|
|
FROM python:3.10-slim |
|
|
|
|
|
|
|
|
RUN apt-get update && apt-get install -y git libgl1-mesa-glx libglib2.0-0 && rm -rf /var/lib/apt/lists/* |
|
|
|
|
|
|
|
|
RUN useradd -m -u 1000 user |
|
|
USER user |
|
|
|
|
|
|
|
|
RUN mkdir -p /home/user/app/hf_cache |
|
|
|
|
|
|
|
|
WORKDIR /home/user/app |
|
|
|
|
|
|
|
|
RUN git clone https://github.com/switchablenorms/CelebAMask-HQ.git |
|
|
|
|
|
|
|
|
ENV PYTHONPATH="/home/user/app/CelebAMask-HQ:/home/user/app/CelebAMask-HQ/face_parsing:${PYTHONPATH}" |
|
|
ENV HF_HOME="/home/user/app/hf_cache" |
|
|
ENV TRANSFORMERS_CACHE="/home/user/app/hf_cache" |
|
|
ENV HUGGINGFACE_HUB_CACHE="/home/user/app/hf_cache" |
|
|
|
|
|
|
|
|
COPY --chown=user:user requirements.txt . |
|
|
COPY --chown=user:user unet.py /home/user/app/CelebAMask-HQ/face_parsing/ |
|
|
COPY --chown=user:user model_utils.py /home/user/app/CelebAMask-HQ/face_parsing/ |
|
|
COPY --chown=user:user utils.py /home/user/app/CelebAMask-HQ/face_parsing/ |
|
|
COPY --chown=user:user app.py . |
|
|
|
|
|
|
|
|
RUN pip install --no-cache-dir -r requirements.txt |
|
|
|
|
|
|
|
|
CMD ["python", "app.py"] |
