Update stf_utils.py

stf_utils.py

@@ -74,10 +74,16 @@ class STFPipeline:
                  template_video_path: str = "templates/front_one_piece_dress_nodded_cut.webm",
                  config_path: str = "front_config.json",
                  checkpoint_path: str = "089.pth",
-                 root_path: str = "works"
+                 #root_path: str = "works"
+                 root_path: str = "/tmp/works"
                  
     ):
-        
+        os.makedirs(root_path, exist_ok=True)
+        import shutil; shutil.copytree('./works', '/tmp/works')
+        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/crop_video_front_one_piece_dress_nodded_cut'
+        import zipfile; zipfile.ZipFile(dir_zip, 'r').extractall(dir_target)
+
         self.config_path = os.path.join(stf_path, config_path)
         self.checkpoint_path = os.path.join(stf_path, checkpoint_path)
         self.work_root_path = os.path.join(stf_path, root_path)