cog.yaml

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

build:
  gpu: true
  system_packages:
    - ffmpeg
  cuda: "11.7"
  python_version: "3.10"
  python_packages:
    - torch==1.13.1
    - torchaudio==0.13.1
    - librosa==0.9.1
    - faster-whisper==0.9.0
    - pydub==0.25.1
    - wavmark
    - eng_to_ipa==0.0.2
    - inflect==7.0.0
    - unidecode==1.3.7
    - whisper-timestamped==1.14.2
    - python-dotenv
    - pypinyin==0.50.0
    - cn2an==0.5.22
    - jieba==0.42.1
    - langid==1.1.6
  run:
    - curl -o /usr/local/bin/pget -L "https://github.com/replicate/pget/releases/download/v0.6.0/pget_linux_x86_64" && chmod +x /usr/local/bin/pget
    - pip install git+https://github.com/myshell-ai/MeloTTS.git
    - python -m unidic download
predict: "predict.py:Predictor"