cog.yaml

build:
  gpu: true
  cuda: "11.3"
  python_version: 3.8
  system_packages:
    - libgl1-mesa-glx
    - libglib2.0-0

  python_packages:
    - imageio==2.22.2
    - matplotlib==3.1.3
    - spacy==3.3.1
    - smplx==0.1.28
    - chumpy==0.70
    - blis==0.7.8
    - click==8.1.3
    - confection==0.0.2
    - ftfy==6.1.1
    - importlib-metadata==5.0.0
    - lxml==4.9.1
    - murmurhash==1.0.8
    - preshed==3.0.7
    - pycryptodomex==3.15.0
    - regex==2022.9.13
    - srsly==2.4.4
    - thinc==8.0.17
    - typing-extensions==4.1.1
    - urllib3==1.26.12
    - wasabi==0.10.1
    - wcwidth==0.2.5

  run:
    - apt update -y && apt-get install ffmpeg -y
#    - python -m spacy download en_core_web_sm
    - git clone https://github.com/openai/CLIP.git sub_modules/CLIP
    - pip install -e sub_modules/CLIP

predict: "sample/predict.py:Predictor"