cog.yaml

build:
  gpu: true
  system_packages:
    - libgl1-mesa-glx
    - libglib2.0-0
    - cmake
    - zip
  python_version: 3.7
  python_packages:
    - torch==1.7.1
    - tensorflow==1.15.0
    - torchvision==0.8.2
    - torchaudio==0.7.2
    - ftfy==5.9
    - regex==2021.4.4
    - tqdm==4.59.0
    - requests==2.25.1
    - matplotlib==3.4.1
    - opencv-python==4.3.0.38
    - dlib==19.18.0
    - scipy==1.6.3
    - "git+git://github.com/openai/CLIP.git@8a665a683d791ed3491fedadcb3c91878f9eb78d"
  pre_install:
    - "mkdir /content"
    - "git clone https://github.com/omertov/encoder4editing.git /content/encoder4editing"
    - "wget https://github.com/ninja-build/ninja/releases/download/v1.8.2/ninja-linux.zip"
    - "unzip ninja-linux.zip -d /usr/local/bin/"
    - "update-alternatives --install /usr/bin/ninja ninja /usr/local/bin/ninja 1 --force"
    - "wget -O /content/shape_predictor_68_face_landmarks.dat.bz2 http://dlib.net/files/shape_predictor_68_face_landmarks.dat.bz2"
    - "cd /content && bzip2 -dk shape_predictor_68_face_landmarks.dat.bz2"
    - "echo > /content/encoder4editing/__init__.py"
    - |
      sed -i 's/img = PIL.Image.open(filepath)/img = PIL.Image.open(filepath).convert(\"RGB\")/' /content/encoder4editing/utils/alignment.py
predict: cog_predict.py:Predictor