Spaces:
Paused
Paused
Update app_celery.py
Browse files- app_celery.py +3 -3
app_celery.py
CHANGED
@@ -194,7 +194,7 @@ def main(args):
|
|
194 |
print('crop_pic_path ',crop_pic_path)
|
195 |
print('crop_info ',crop_info)
|
196 |
|
197 |
-
if
|
198 |
print("Can't get the coeffs of the input")
|
199 |
return
|
200 |
|
@@ -221,13 +221,13 @@ def main(args):
|
|
221 |
ref_pose_coeff_path=None
|
222 |
print('ref_eyeblink_coeff_path',ref_pose_coeff_path)
|
223 |
|
224 |
-
batch = get_data(
|
225 |
coeff_path = audio_to_coeff.generate(batch, save_dir, pose_style, ref_pose_coeff_path)
|
226 |
|
227 |
|
228 |
if args.face3dvis:
|
229 |
from src.face3d.visualize import gen_composed_video
|
230 |
-
gen_composed_video(args, device,
|
231 |
|
232 |
# data = get_facerender_data(coeff_path, crop_pic_path, first_coeff_path, audio_path,
|
233 |
# batch_size, input_yaw_list, input_pitch_list, input_roll_list,
|
|
|
194 |
print('crop_pic_path ',crop_pic_path)
|
195 |
print('crop_info ',crop_info)
|
196 |
|
197 |
+
if first_coeff_path is None:
|
198 |
print("Can't get the coeffs of the input")
|
199 |
return
|
200 |
|
|
|
221 |
ref_pose_coeff_path=None
|
222 |
print('ref_eyeblink_coeff_path',ref_pose_coeff_path)
|
223 |
|
224 |
+
batch = get_data(first_coeff_path, audio_path, device, ref_eyeblink_coeff_path, still=args.still)
|
225 |
coeff_path = audio_to_coeff.generate(batch, save_dir, pose_style, ref_pose_coeff_path)
|
226 |
|
227 |
|
228 |
if args.face3dvis:
|
229 |
from src.face3d.visualize import gen_composed_video
|
230 |
+
gen_composed_video(args, device, first_coeff_path, coeff_path, audio_path, os.path.join(save_dir, '3dface.mp4'))
|
231 |
|
232 |
# data = get_facerender_data(coeff_path, crop_pic_path, first_coeff_path, audio_path,
|
233 |
# batch_size, input_yaw_list, input_pitch_list, input_roll_list,
|