Spaces:
Paused
Paused
| # Configuration for Cog ⚙️ | |
| # Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md | |
| build: | |
| gpu: true | |
| system_packages: | |
| - "libgl1-mesa-glx" | |
| - "libglib2.0-0" | |
| - "ffmpeg" | |
| python_version: "3.11" | |
| python_packages: | |
| - "torch==2.0.1" | |
| - "torchvision==0.15.2" | |
| - "basicsr==1.4.2" | |
| - "kornia==0.5.1" | |
| - "face-alignment==1.3.4" | |
| - "ninja==1.10.2.3" | |
| - "einops==0.4.1" | |
| - "facexlib==0.2.5" | |
| - "librosa==0.9.2" | |
| - "cmake==3.27.7" | |
| - "numpy==1.23.4" | |
| run: | |
| - pip install dlib | |
| - mkdir -p /root/.pyenv/versions/3.11.6/lib/python3.11/site-packages/facexlib/weights/ && wget --output-document "/root/.pyenv/versions/3.11.6/lib/python3.11/site-packages/facexlib/weights/detection_Resnet50_Final.pth" "https://github.com/xinntao/facexlib/releases/download/v0.1.0/detection_Resnet50_Final.pth" | |
| - mkdir -p /root/.pyenv/versions/3.11.6/lib/python3.11/site-packages/facexlib/weights/ && wget --output-document "/root/.pyenv/versions/3.11.6/lib/python3.11/site-packages/facexlib/weights/parsing_parsenet.pth" "https://github.com/xinntao/facexlib/releases/download/v0.2.2/parsing_parsenet.pth" | |
| - mkdir -p /root/.cache/torch/hub/checkpoints/ && wget --output-document "/root/.cache/torch/hub/checkpoints/s3fd-619a316812.pth" "https://www.adrianbulat.com/downloads/python-fan/s3fd-619a316812.pth" | |
| - mkdir -p /root/.cache/torch/hub/checkpoints/ && wget --output-document "/root/.cache/torch/hub/checkpoints/2DFAN4-cd938726ad.zip" "https://www.adrianbulat.com/downloads/python-fan/2DFAN4-cd938726ad.zip" | |
| predict: "predict.py:Predictor" | |