| # Configuration for Cog ⚙️ | |
| # Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md | |
| build: | |
| # set to true if your model requires a GPU | |
| cuda: "10.2" | |
| gpu: true | |
| # a list of ubuntu apt packages to install | |
| system_packages: | |
| # - "libgl1-mesa-glx" | |
| # - "libglib2.0-0" | |
| - "libgl1-mesa-dev" | |
| # python version in the form '3.8' or '3.8.12' | |
| python_version: "3.8" | |
| # a list of packages in the format <package-name>==<version> | |
| python_packages: | |
| # - "numpy==1.19.4" | |
| # - "torch==1.8.0" | |
| # - "torchvision==0.9.0" | |
| - "numpy==1.23.1" | |
| - "torch==1.8.0" | |
| - "torchvision==0.9.0" | |
| - "opencv-python==4.6.0.66" | |
| - "pandas==1.4.3" | |
| - "pillow==9.2.0" | |
| - "tqdm==4.64.0" | |
| - "scikit-image==0.19.3" | |
| - "scikit-learn==1.1.2" | |
| - "scipy==1.9.1" | |
| # commands run after the environment is setup | |
| # run: | |
| # - "echo env is ready!" | |
| # - "echo another command if needed" | |
| # predict.py defines how predictions are run on your model | |
| predict: "predict.py:Predictor" | |
| #image: "r8.im/menghanxia/disco" |