| # Configuration for Cog ⚙️ | |
| # Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md | |
| build: | |
| gpu: true | |
| system_packages: | |
| - "libgl1-mesa-glx" | |
| - "libglib2.0-0" | |
| - "cmake" | |
| python_version: "3.11" | |
| python_packages: | |
| - torch==2.0.1 | |
| - torchvision==0.15.2 | |
| - lmdb==1.4.1 | |
| - opencv_python==4.9.0.80 | |
| - Pillow==10.1.0 | |
| - PyYAML==6.0.1 | |
| - scipy==1.11.4 | |
| - timm==0.9.12 | |
| - tqdm==4.65.0 | |
| - scikit-image==0.22.0 | |
| - ipython | |
| run: | |
| - pip install dlib | |
| predict: "predict.py:Predictor" | |