Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
1510e36
1
Parent(s):
877d523
Update webgui.py
Browse files
webgui.py
CHANGED
@@ -129,7 +129,7 @@ face_locator.load_state_dict(torch.load(config.face_locator_path, map_location='
|
|
129 |
audio_processor = load_audio_model(model_path=config.audio_model_path, device=device)
|
130 |
|
131 |
## load face detector params
|
132 |
-
face_detector = MTCNN(image_size=320, margin=0, min_face_size=20, thresholds=[0.6, 0.7, 0.7], factor=0.709, post_process=True, device=
|
133 |
|
134 |
############# model_init finished #############
|
135 |
|
|
|
129 |
audio_processor = load_audio_model(model_path=config.audio_model_path, device=device)
|
130 |
|
131 |
## load face detector params
|
132 |
+
face_detector = MTCNN(image_size=320, margin=0, min_face_size=20, thresholds=[0.6, 0.7, 0.7], factor=0.709, post_process=True, device="cpu")
|
133 |
|
134 |
############# model_init finished #############
|
135 |
|