multimodalart HF staff commited on
Commit
1510e36
1 Parent(s): 877d523

Update webgui.py

Browse files
Files changed (1) hide show
  1. webgui.py +1 -1
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=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