add debug log

webgui.py

@@ -164,6 +164,8 @@ lmk_extractor = LMKExtractor()
 def process_video(uploaded_img, uploaded_audio, width, height, length, facemask_dilation_ratio, facecrop_dilation_ratio, context_frames, context_overlap, cfg, steps, sample_rate, fps, device):
     #### face musk prepare
     face_img = cv2.imread(uploaded_img)
+    if face_img is None:
+        raise gr.Error("input image should be uploaded or selected.")
     face_mask = np.zeros((face_img.shape[0], face_img.shape[1])).astype('uint8')
     det_bboxes, probs = face_detector.detect(face_img)
     select_bbox = select_face(det_bboxes, probs)
@@ -185,8 +187,7 @@ def process_video(uploaded_img, uploaded_audio, width, height, length, facemask_
         face_mask = crop_and_pad(face_mask, crop_rect)
         face_img = cv2.resize(face_img, (width, height))
         face_mask = cv2.resize(face_mask, (width, height))
-
-
+    print('face detect done.')
     # ==================== face_locator =====================
     '''
     driver_video = "./assets/driven_videos/c.mp4"
@@ -235,6 +236,7 @@ def process_video(uploaded_img, uploaded_audio, width, height, length, facemask_
         fps=fps,
         context_overlap=context_overlap
     ).videos
+    print('video pipe done.')
 
     save_dir = Path("output/tmp")
     save_dir.mkdir(exist_ok=True, parents=True)