cog.yaml

# Configuration for Cog ⚙️
# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md

image: r8.im/vishjain/wav2lip

build:
  # set to true if your model requires a GPU
  gpu: true

  # a list of ubuntu apt packages to install
  system_packages:
     - ffmpeg
     - cmake

  # python version in the form '3.11' or '3.11.4'
  python_version: "3.8"

  # a list of packages in the format <package-name>==<version>
  python_packages:
    - numpy==1.23.4
    - librosa==0.7.0
    - torch==1.12.1
    - torchvision==0.13.1
    - opencv-python==4.6.0.66
    - tqdm==4.45.0
    - numba==0.48
    - mediapipe==0.8.11
    - ffmpeg-python

  # commands run after the environment is setup
  run:
    - mkdir -p /src/checkpoints/ && wget --output-document "/src/checkpoints/mobilenet.pth" "https://github.com/elliottzheng/face-detection/releases/download/0.0.1/mobilenet0.25_Final.pth"
    - mkdir -p /src/checkpoints/ && wget --output-document "/src/checkpoints/wav2lip_gan.pth" "https://iiitaphyd-my.sharepoint.com/personal/radrabha_m_research_iiit_ac_in/_layouts/15/download.aspx?share=EdjI7bZlgApMqsVoEUUXpLsBxqXbn5z8VTmoxp55YNDcIA"
    # - mkdir -p /src/checkpoints/ && wget -O /src/checkpoints/mobilenet.pth 'https://github.com/elliottzheng/face-detection/releases/download/0.0.1/mobilenet0.25_Final.pth' && wget -O /src/checkpoints/wav2lip_gan.pth 'https://iiitaphyd-my.sharepoint.com/personal/radrabha_m_research_iiit_ac_in/_layouts/15/download.aspx?share=EdjI7bZlgApMqsVoEUUXpLsBxqXbn5z8VTmoxp55YNDcIA'
    - pip install git+https://github.com/elliottzheng/batch-face.git@master
    - echo "Environment is ready!"

# predict.py defines how predictions are run on your model
predict: "predict.py:Predictor"