salomonsky commited on
Commit
fd1606e
1 Parent(s): 1345bfc

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +6 -2
inference.py CHANGED
@@ -9,6 +9,12 @@ from models import Wav2Lip
9
  import platform
10
  import cv2
11
 
 
 
 
 
 
 
12
  parser = argparse.ArgumentParser(description='Inference code to lip-sync videos in the wild using Wav2Lip models')
13
 
14
  parser.add_argument('--checkpoint_path', type=str,
@@ -126,8 +132,6 @@ def face_detect(images):
126
  del detector
127
  return results,head_exist
128
 
129
- import cv2
130
-
131
  def datagen(frames, mels):
132
  img_batch,head_exist_batch, mel_batch, frame_batch, coords_batch = [], [], [], [],[]
133
 
 
9
  import platform
10
  import cv2
11
 
12
+ try:
13
+ last_face = cv2.imread("last_face.jpg")
14
+ except Exception as e:
15
+ print(f"No se pudo cargar 'last_face.jpg': {e}")
16
+ last_face = None
17
+
18
  parser = argparse.ArgumentParser(description='Inference code to lip-sync videos in the wild using Wav2Lip models')
19
 
20
  parser.add_argument('--checkpoint_path', type=str,
 
132
  del detector
133
  return results,head_exist
134
 
 
 
135
  def datagen(frames, mels):
136
  img_batch,head_exist_batch, mel_batch, frame_batch, coords_batch = [], [], [], [],[]
137