Update videoretalking/inference_function.py

videoretalking/inference_function.py

@@ -4,6 +4,7 @@ from tqdm import tqdm
 from PIL import Image
 from scipy.io import loadmat
 from moviepy.editor import AudioFileClip, VideoFileClip
+import shutil
 
 sys.path.insert(0, 'third_part')
 sys.path.insert(0, 'third_part/GPEN')
@@ -278,6 +279,9 @@ def video_lipsync_correctness(face, audio_path, face3d_net_path, outfile=None, t
     # Write the result to the output file
     video_clip.write_videofile(outfile, codec='libx264', audio_codec='aac')
     print('outfile:', outfile)
+    if os.path.isdir(tmp_dir):
+        shutil.rmtree(tmp_dir)
+        print(f'Cleaned up temporary directory: {tmp_dir}')
 
 # frames:256x256, full_frames: original size
 def datagen(frames, mels, full_frames, frames_pil, cox, face, static, LNet_batch_size, img_size):