Update stf_utils.py

stf_utils.py

@@ -79,28 +79,29 @@ class STFPipeline:
                  checkpoint_path: str = "089.pth",
                  #root_path: str = "works"
                  root_path: str = "/tmp/works",
-                 male : bool = False
+                 female_video: bool=True
                  
     ):
         #os.makedirs(root_path, exist_ok=True)
         shutil.copytree('/home/user/app/stf/works', '/tmp/works', dirs_exist_ok=True)
         
+        
 
-        if not male:
-            dir_zip='/tmp/works/preprocess/nasilhong_f_v1_front/crop_video_front_one_piece_dress_nodded_cut.zip'
-            dir_target='/tmp/works/preprocess/nasilhong_f_v1_front/'
+        if female_video:
+            dir_zip= os.path.join(root_path, 'preprocess/nasilhong_f_v1_front/crop_video_front_one_piece_dress_nodded_cut.zip')
+            dir_target=os.path.join(root_path,'preprocess/nasilhong_f_v1_front/')
             zipfile.ZipFile(dir_zip, 'r').extractall(dir_target)
             
-            dir_zip='/tmp/works/preprocess/nasilhong_f_v1_front/front_one_piece_dress_nodded_cut.zip'
-            dir_target='/tmp/works/preprocess/nasilhong_f_v1_front/'
+            dir_zip=os.path.join(root_path,'preprocess/nasilhong_f_v1_front/front_one_piece_dress_nodded_cut.zip')
+            dir_target=os.path.join(root_path,'preprocess/nasilhong_f_v1_front/')
             zipfile.ZipFile(dir_zip, 'r').extractall(dir_target)
         else:
-            dir_zip='/tmp/works/preprocess/Ian_v3_front/crop_video_Cam2_2309071202_0012_Natural_Looped.zip'
-            dir_target='/tmp/works/preprocess/Ian_v3_front/'
+            dir_zip= os.path.join(root_path, 'preprocess/Ian_v3_front/crop_video_Cam2_2309071202_0012_Natural_Looped.zip')
+            dir_target=os.path.join(root_path,'preprocess/Ian_v3_front/')
             zipfile.ZipFile(dir_zip, 'r').extractall(dir_target)
             
-            dir_zip='/tmp/works/preprocess/Ian_v3_front/Cam2_2309071202_0012_Natural_Looped.zip'
-            dir_target='/tmp/works/preprocess/Ian_v3_front/'
+            dir_zip=os.path.join(root_path,'preprocess/Ian_v3_front/Cam2_2309071202_0012_Natural_Looped.zip')
+            dir_target=os.path.join(root_path,'preprocess/Ian_v3_front/')
             zipfile.ZipFile(dir_zip, 'r').extractall(dir_target)