From e639721d11f9618df252c432236f293ddd698fae Mon Sep 17 00:00:00 2001
From: AIFSH <149982694+AIFSH@users.noreply.github.com>
Date: Fri, 31 May 2024 21:45:42 +0800
Subject: [PATCH] Update utils.py

---
 V_Express/pipelines/utils.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/V_Express/pipelines/utils.py b/V_Express/pipelines/utils.py
index b9207eb..23120d2 100644
--- a/V_Express/pipelines/utils.py
+++ b/V_Express/pipelines/utils.py
@@ -86,8 +86,7 @@ def save_video(video_tensor, audio_path, output_path, fps=30.0):
     cmd = (f'{get_ffmpeg_exe()} -i "{temp_output_path}" -i "{audio_path}" '
            f'-map 0:v -map 1:a -c:v h264 -shortest -y "{output_path}" -loglevel quiet')
     os.system(cmd)
-    import shutil
-    shutil.remove(temp_output_path)
+    os.remove(temp_output_path)
     # os.system(f'rm -rf "{temp_output_path}"')