Spaces:
Runtime error
Runtime error
saimemrekanat
commited on
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'),
|