Update app.py

app.py

@@ -166,14 +166,14 @@ def morph_faces(img1, img2, alpha, dim, step):
     return (out*255).astype(np.uint8)
 
 
-def process_video(video_path, ref_img, trans, res, step, feather):
+def process_video(video_path, ref_img, trans, res, step, feather, padding):
     cap = cv2.VideoCapture(video_path)
     fps = cap.get(cv2.CAP_PROP_FPS) or 24
     total = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
 
     # Prepare masked reference
     ref_bgr = cv2.cvtColor(ref_img, cv2.COLOR_RGB2BGR)
-    mask_ref, ref_box = get_face_mask_box(ref_bgr, feather)
+    mask_ref, ref_box = get_face_mask_box(ref_bgr, feather, padding)
     if mask_ref is None:
         return None, None, None, None
     x_r, y_r, w_r, h_r = ref_box
@@ -196,7 +196,7 @@ def process_video(video_path, ref_img, trans, res, step, feather):
     for i in range(total):
         ret, frame = cap.read()
         if not ret: break
-        mask_roi, box = get_face_mask_box(frame, feather)
+        mask_roi, box = get_face_mask_box(frame, feather, padding)
         if mask_roi is None:
             out_vid.write(frame)
             continue