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

fix window error
---
 V_Express/pipelines/utils.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/V_Express/pipelines/utils.py b/V_Express/pipelines/utils.py
index 02ba1a4..10a0e1f 100644
--- a/V_Express/pipelines/utils.py
+++ b/V_Express/pipelines/utils.py
@@ -86,7 +86,9 @@ 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)
-    os.system(f'rm -rf "{temp_output_path}"')
+    import shutil
+    shutil.rmtree(temp_output_path)
+    # os.system(f'rm -rf "{temp_output_path}"')
 
 
 def compute_dist(x1, y1, x2, y2):