build:
  python_version: "3.8"

  gpu: true

  python_packages:
    - 'huggingface_hub==0.8.1'
    - 'torch==1.7.1'
    - 'torchvision==0.8.2'
    - 'tokenizers==0.10.2'
    - 'pyflakes==2.2.0'
    - 'tqdm==4.46.0'
    - 'pytorch-lightning==1.5'
    - 'einops==0.4.1'
    - 'omegaconf==2.2.2'
    - 'git+https://github.com/openai/CLIP.git'
    - 'numpy==1.23.1'
    - 'pillow==9.2.0'
    - 'python-dotenv==0.20.0'

predict: "predict.py:Predictor"