cog.yaml

build:
  cuda: "10.2"
  gpu: true
  python_version: "3.8"
  system_packages:
    - "libgl1-mesa-glx"
    - "libglib2.0-0"
  python_packages:
    - "numpy==1.20.1"
    - "torch==1.10.1"
    - "lmdb==1.3.0"
    - "scikit-learn==1.1.1"
    - "opencv-python==4.5.5.64"
    - "einops==0.4.1"
    - "torchvision==0.11.2"
    - "tb-nightly==2.10.0a20220801"
    - "yapf==0.32.0"
    - "pyyaml==6.0"
    - "tqdm==4.63.0"
    - "scipy==1.8.1"
    - "timm==0.6.5"

  run:
   - pip install basicsr
   - pip install facexlib
   - pip install realesrgan
   - mkdir -p /root/.pyenv/versions/3.8.13/lib/python3.8/site-packages/facexlib/weights/ && wget --output-document "/root/.pyenv/versions/3.8.13/lib/python3.8/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.8.13/lib/python3.8/site-packages/facexlib/weights/ && wget --output-document "/root/.pyenv/versions/3.8.13/lib/python3.8/site-packages/facexlib/weights/parsing_parsenet.pth" "https://github.com/xinntao/facexlib/releases/download/v0.2.2/parsing_parsenet.pth"

predict: "predict.py:Predictor"