zejunyang commited on
Commit
9600e7d
1 Parent(s): ac336de
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -234,7 +234,7 @@ def video2video(ref_img, source_video, size=512, steps=25, length=60, seed=42):
234
  save_dir.mkdir(exist_ok=True, parents=True)
235
 
236
  ref_image_np = cv2.cvtColor(ref_img, cv2.COLOR_RGB2BGR)
237
- # ref_image_np = crop_face(ref_image_np, lmk_extractor)
238
  if ref_image_np is None:
239
  return None, Image.fromarray(ref_img)
240
 
 
234
  save_dir.mkdir(exist_ok=True, parents=True)
235
 
236
  ref_image_np = cv2.cvtColor(ref_img, cv2.COLOR_RGB2BGR)
237
+ ref_image_np = crop_face(ref_image_np, lmk_extractor)
238
  if ref_image_np is None:
239
  return None, Image.fromarray(ref_img)
240