cog.yaml

build:
  python_version: "3.6"
  gpu: false
  python_packages:
    - future==0.18.2
    - numpy==1.19.5
    - librosa==0.8.1
    - soundfile==0.10.3.post1
    - musdb==0.4.0
    - museval==0.4.0
    - h5py==3.1.0
    - tqdm==4.62.1
    - torch==1.4.0
    - torchvision==0.5.0
    - tensorboard==2.6.0
    - sortedcontainers==2.4.0
  system_packages:
    - libsndfile-dev
    - ffmpeg
predict: "cog_predict.py:waveunetPredictor"