Spanicin commited on
Commit
30ccf11
·
verified ·
1 Parent(s): c1bcc37

Update app_celery.py

Browse files
Files changed (1) hide show
  1. 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 first_coeff_new_path is None:
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(first_coeff_new_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_new_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,
 
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,