Spaces:
Runtime error
Runtime error
saimemrekanat
commited on
Commit
•
6750432
1
Parent(s):
3e63f01
face detect test
Browse files- vtoonify_model.py +2 -1
vtoonify_model.py
CHANGED
@@ -52,8 +52,9 @@ class Model():
|
|
52 |
}
|
53 |
|
54 |
self.landmarkpredictor = self._create_dlib_landmark_model()
|
|
|
55 |
self.parsingpredictor = self._create_parsing_model()
|
56 |
-
self.pspencoder = self._load_encoder()
|
57 |
self.transform = transforms.Compose([
|
58 |
transforms.ToTensor(),
|
59 |
transforms.Normalize(mean=[0.5, 0.5, 0.5],std=[0.5,0.5,0.5]),
|
|
|
52 |
}
|
53 |
|
54 |
self.landmarkpredictor = self._create_dlib_landmark_model()
|
55 |
+
self.cnn_model = self._create_dlib_landmark_model()
|
56 |
self.parsingpredictor = self._create_parsing_model()
|
57 |
+
self.pspencoder = self._load_encoder()
|
58 |
self.transform = transforms.Compose([
|
59 |
transforms.ToTensor(),
|
60 |
transforms.Normalize(mean=[0.5, 0.5, 0.5],std=[0.5,0.5,0.5]),
|