52Hz commited on
Commit
d2203e3
1 Parent(s): 586d6c2

Create cog.yaml

Browse files
Files changed (1) hide show
  1. cog.yaml +20 -0
cog.yaml ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ build:
2
+ gpu: true
3
+ python_version: "3.8"
4
+ system_packages:
5
+ - "libgl1-mesa-glx"
6
+ - "libglib2.0-0"
7
+ python_packages:
8
+ - "torchvision==0.9.0"
9
+ - "torch==1.8.0"
10
+ - "numpy==1.19.4"
11
+ - "opencv-python==4.5.1.48"
12
+ - "scikit-image==0.18.1"
13
+ - "natsort==7.1.1"
14
+ - "opencv==4.0.1"
15
+ - "tqdm==4.62.2"
16
+ - "Pillow==8.3.2"
17
+ - "timm==0.4.12"
18
+ - "ipython==7.19.0"
19
+
20
+ predict: "predict.py:Predictor"