Spanicin commited on
Commit
3dc1e9c
·
verified ·
1 Parent(s): 49b64d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -130,10 +130,10 @@ def process_chunk(audio_chunk, args):
130
  torch.cuda.empty_cache()
131
 
132
  if args.ref_pose is not None:
133
- ref_pose_videoname = os.path.splitext(os.path.split(ref_pose)[-1])[0]
134
  ref_pose_frame_dir = os.path.join(save_dir, ref_pose_videoname)
135
  os.makedirs(ref_pose_frame_dir, exist_ok=True)
136
- ref_pose_coeff_path, _, _ = preprocess_model.generate(ref_pose, ref_pose_frame_dir)
137
  print('ref_eyeblink_coeff_path',ref_pose_coeff_path)
138
  else:
139
  ref_pose_coeff_path = None
 
130
  torch.cuda.empty_cache()
131
 
132
  if args.ref_pose is not None:
133
+ ref_pose_videoname = os.path.splitext(os.path.split(args.ref_pose)[-1])[0]
134
  ref_pose_frame_dir = os.path.join(save_dir, ref_pose_videoname)
135
  os.makedirs(ref_pose_frame_dir, exist_ok=True)
136
+ ref_pose_coeff_path, _, _ = preprocess_model.generate(args.ref_pose, ref_pose_frame_dir)
137
  print('ref_eyeblink_coeff_path',ref_pose_coeff_path)
138
  else:
139
  ref_pose_coeff_path = None