| build: | |
| gpu: true | |
| cuda: "12.1" | |
| python_version: "3.10" | |
| system_packages: | |
| - "libgl1-mesa-glx" | |
| - "libglib2.0-0" | |
| python_packages: | |
| - "torch==2.3.0" | |
| - "torchvision" | |
| - "transformers" | |
| - "opencv-python" | |
| - "pillow" | |
| - "numpy" | |
| - "scipy" | |
| - "huggingface_hub" | |
| image: "r8.im/bawolf/break-video-clip" | |
| predict: "predict.py:Predictor" |