Spaces:
Runtime error
Runtime error
Commit
·
8ecd0a6
1
Parent(s):
f1a45cb
face detect test
Browse files- vtoonify_model.py +3 -3
vtoonify_model.py
CHANGED
|
@@ -73,9 +73,9 @@ class Model():
|
|
| 73 |
|
| 74 |
@staticmethod
|
| 75 |
def _create_dlib_landmark_cnn_model():
|
| 76 |
-
return dlib.
|
| 77 |
-
|
| 78 |
-
|
| 79 |
def _create_parsing_model(self):
|
| 80 |
parsingpredictor = BiSeNet(n_classes=19)
|
| 81 |
parsingpredictor.load_state_dict(torch.load(huggingface_hub.hf_hub_download(MODEL_REPO, 'models/faceparsing.pth'),
|
|
|
|
| 73 |
|
| 74 |
@staticmethod
|
| 75 |
def _create_dlib_landmark_cnn_model():
|
| 76 |
+
return dlib.cnn_face_detection_model_v1(huggingface_hub.hf_hub_download(MODEL_REPO,
|
| 77 |
+
'models/mmod_human_face_detector.dat'))
|
| 78 |
+
|
| 79 |
def _create_parsing_model(self):
|
| 80 |
parsingpredictor = BiSeNet(n_classes=19)
|
| 81 |
parsingpredictor.load_state_dict(torch.load(huggingface_hub.hf_hub_download(MODEL_REPO, 'models/faceparsing.pth'),
|