Spaces:
Runtime error
Runtime error
Update modules/sadtalker_test.py
Browse files
modules/sadtalker_test.py
CHANGED
|
@@ -86,7 +86,7 @@ class SadTalker():
|
|
| 86 |
batch = get_data(first_coeff_path, audio_path, self.device)
|
| 87 |
coeff_path = self.audio_to_coeff.generate(batch, save_dir, pose_style)
|
| 88 |
#coeff2video
|
| 89 |
-
batch_size =
|
| 90 |
data = get_facerender_data(coeff_path, crop_pic_path, first_coeff_path, audio_path, batch_size, still_mode=still_mode)
|
| 91 |
self.animate_from_coeff.generate(data, save_dir, enhancer='gfpgan' if use_enhancer else None)
|
| 92 |
video_name = data['video_name']
|
|
|
|
| 86 |
batch = get_data(first_coeff_path, audio_path, self.device)
|
| 87 |
coeff_path = self.audio_to_coeff.generate(batch, save_dir, pose_style)
|
| 88 |
#coeff2video
|
| 89 |
+
batch_size = 8
|
| 90 |
data = get_facerender_data(coeff_path, crop_pic_path, first_coeff_path, audio_path, batch_size, still_mode=still_mode)
|
| 91 |
self.animate_from_coeff.generate(data, save_dir, enhancer='gfpgan' if use_enhancer else None)
|
| 92 |
video_name = data['video_name']
|