Spaces:
Paused
Paused
Update app.py
Browse files
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
|