Update app.py

app.py

@@ -184,7 +184,7 @@ def main(args):
     print('crop_pic_path ',crop_pic_new_path)
     print('crop_info ',crop_info)
 
-    if first_coeff_path is None:
+    if first_coeff_new_path is None:
         print("Can't get the coeffs of the input")
         return
 
@@ -211,20 +211,20 @@ def main(args):
         ref_pose_coeff_path=None
         print('ref_eyeblink_coeff_path',ref_pose_coeff_path)
 
-    batch = get_data(first_coeff_path, audio_path, device, ref_eyeblink_coeff_path, still=args.still)
+    batch = get_data(first_coeff_new_path, audio_path, device, ref_eyeblink_coeff_path, still=args.still)
     coeff_path = audio_to_coeff.generate(batch, save_dir, pose_style, ref_pose_coeff_path)
 
 
     if args.face3dvis:
         from src.face3d.visualize import gen_composed_video
-        gen_composed_video(args, device, first_coeff_path, coeff_path, audio_path, os.path.join(save_dir, '3dface.mp4'))
+        gen_composed_video(args, device, first_coeff_new_path, coeff_path, audio_path, os.path.join(save_dir, '3dface.mp4'))
   
     # data = get_facerender_data(coeff_path, crop_pic_path, first_coeff_path, audio_path,
                                 # batch_size, input_yaw_list, input_pitch_list, input_roll_list,
                                 # expression_scale=args.expression_scale, still_mode=args.still, preprocess=args.preprocess, size=args.size)
 
 
-    data = get_facerender_data(coeff_path, crop_pic_path, first_coeff_path, audio_path, 
+    data = get_facerender_data(coeff_path, crop_pic_new_path, first_coeff_new_path, audio_path, 
                                 batch_size, input_yaw_list, input_pitch_list, input_roll_list,
                                 expression_scale=args.expression_scale, still_mode=args.still, preprocess=args.preprocess)