Skip to content

Commit

Permalink
fix typo in inference.py
Browse files Browse the repository at this point in the history
  • Loading branch information
suravshresth committed Oct 19, 2023
1 parent a0796c6 commit d8d2789
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -149,7 +149,7 @@ def main():

if not os.path.isfile('temp/'+base_name+'_stablized.npy') or args.re_preprocess:
imgs = []
for idx in tqdm(range(len(frames_pil)), desc="[Step 3] Stablize the expression In Video:"):
for idx in tqdm(range(len(frames_pil)), desc="[Step 3] Stabilize the expression In Video:"):
if args.one_shot:
source_img = trans_image(frames_pil[0]).unsqueeze(0).to(device)
semantic_source_numpy = semantic_npy[0:1]
Expand Down

0 comments on commit d8d2789

Please sign in to comment.