cog.yaml

build:
  cuda: "11.0"
  gpu: true
  python_version: "3.7"
  system_packages:
    - "libgl1-mesa-glx"
    - "libglib2.0-0"
  python_packages:
    - "ipython==7.21.0"
    - "transformers==4.19.2"
    - "h5py==3.7.0"
    - "numpy==1.20.3"
    - "pandas==1.3.3"
    - "scikit-image==0.18.3"
    - "ipywidgets==7.7.0"
    - "wandb==0.12.17"
    - "bert-score==0.3.11"
    - "ftfy==6.1.1"
    - "timm==0.5.4"
    - "lmdbdict==0.2.2"
    - "yacs==0.1.8"
    - "pyemd==0.5.1"
    - "gensim==4.2.0"
    - "pytorch-lightning==1.6.3"

predict: "predict.py:Predictor"