Spaces:
Running
Running
narugo1992
commited on
Commit
•
2d30881
1
Parent(s):
89047b5
dev(narugo): append new models
Browse files
censor.py
CHANGED
@@ -9,6 +9,7 @@ from plot import detection_visualize
|
|
9 |
from yolo_ import _image_preprocess, _data_postprocess
|
10 |
|
11 |
_CENSOR_MODELS = [
|
|
|
12 |
'censor_detect_v0.7_s',
|
13 |
]
|
14 |
_DEFAULT_CENSOR_MODEL = _CENSOR_MODELS[0]
|
|
|
9 |
from yolo_ import _image_preprocess, _data_postprocess
|
10 |
|
11 |
_CENSOR_MODELS = [
|
12 |
+
'censor_detect_v0.8_s',
|
13 |
'censor_detect_v0.7_s',
|
14 |
]
|
15 |
_DEFAULT_CENSOR_MODEL = _CENSOR_MODELS[0]
|