Update stf_utils.py

stf_utils.py

@@ -13,6 +13,9 @@ from rich.progress import track
 from tqdm import tqdm
 
 import stf_alternative
+import os.path as osp
+import shutil
+import zipfile
 
 
 
@@ -80,9 +83,8 @@ class STFPipeline:
                  
     ):
         #os.makedirs(root_path, exist_ok=True)
-        import shutil; shutil.copytree('/home/user/app/stf/works', '/tmp/works', dirs_exist_ok=True)
+        shutil.copytree('/home/user/app/stf/works', '/tmp/works', dirs_exist_ok=True)
         
-        import zipfile
 
         if not male:
             dir_zip='/tmp/works/preprocess/nasilhong_f_v1_front/crop_video_front_one_piece_dress_nodded_cut.zip'
@@ -189,5 +191,8 @@ class STFPipeline:
                 pass
 
         images2video(results, save_path)
+
+        save_path_aud = save_path.replace('.mp4', '_aud.mp4')
+        merge_audio_video(save_path, audio, save_path_aud)
                                 
-        return save_path
+        return save_path_aud #save_path