build:
  gpu: true
  python_version: "3.8"
  system_packages:
    - "libgl1-mesa-glx"
    - "libglib2.0-0"
  python_packages:
    - "Pillow==8.4.0"
    - "iPython==7.22.0"
    - "numpy==1.19.2"
    - "torch==1.7.1"
    - "torchvision==0.8.2"
    - "opencv-python==4.5.4.60"
    - "regex==2021.11.10"
    - "ftfy==6.0.3"
    - "tqdm==4.62.3"
  run:
    - pip install git+https://github.com/openai/CLIP.git

predict: "predict.py:Predictor"