| |
| 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"] |
