cog.yaml

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

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

  # a list of ubuntu apt packages to install
  system_packages:
    - "aria2"
    - "libgl1-mesa-glx"
    - "zip"
  #   - "libglib2.0-0"

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

  # a list of packages in the format <package-name>==<version>
  python_packages:
    - "torch"
    - "torchvision"
    - "torchaudio"
    - "yt-dlp==2023.7.6"
    - "ffmpeg==1.4"
    - "demucs>=4.0.1"
    - "numpy==1.23.5"
    - "librosa==0.9.1"
    - "soundfile==0.12.1"
    - "joblib>=1.1.0"
    - "numba==0.56.4"
    - "numpy==1.23.5"
    - "scipy==1.11.1"
    - "librosa==0.9.1"
    - "llvmlite==0.39.0"
    - "fairseq==0.12.2"
    - "faiss-cpu==1.7.3"
    - "gradio==3.34.0"
    - "Cython==3.0.5"
    - "pydub>=0.25.1"
    - "soundfile>=0.12.1"
    - "ffmpeg-python>=0.2.0"
    - "tensorboardX"
    - "Jinja2>=3.1.2"
    - "json5==0.9.14"
    - "Markdown==3.5.1"
    - "matplotlib>=3.7.0"
    - "matplotlib-inline>=0.1.3"
    - "praat-parselmouth>=0.4.2"
    - "Pillow>=9.1.1"
    - "resampy>=0.4.2"
    - "scikit-learn==1.3.2"
    - "tensorboardX==2.6.2.2"
    - "tqdm>=4.63.1"
    - "tornado>=6.1"
    - "Werkzeug>=2.2.3"
    - "uc-micro-py>=1.0.1"
    - "sympy>=1.11.1"
    - "tabulate>=0.8.10"
    - "PyYAML>=6.0"
    - "pyasn1>=0.4.8"
    - "pyasn1-modules>=0.2.8"
    - "fsspec>=2022.11.0"
    - "absl-py>=1.2.0"
    - "audioread==3.0.1"
    - "uvicorn>=0.21.1"
    - "colorama>=0.4.5"
    - "pyworld==0.3.2"
    - "httpx==0.25.1"
    - "onnxruntime; sys_platform == 'darwin'"
    - "onnxruntime-gpu; sys_platform != 'darwin'"
    - "torchcrepe==0.0.20"
    - "fastapi==0.88"
    - "ffmpy==0.3.1"
    - "python-dotenv>=1.0.0"
    - "av==11.0.0"

  # commands run after the environment is setup
  run:
    - pip install --upgrade pip
    - apt-get update && apt-get install -y ffmpeg
    - pip install imageio[ffmpeg]
    - pip install demucs

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