Spaces:
Running
on
Zero
Running
on
Zero
Update stf_utils.py
Browse files- stf_utils.py +7 -0
stf_utils.py
CHANGED
@@ -80,9 +80,16 @@ class STFPipeline:
|
|
80 |
):
|
81 |
#os.makedirs(root_path, exist_ok=True)
|
82 |
import shutil; shutil.copytree('/home/user/app/stf/works', '/tmp/works', dirs_exist_ok=True)
|
|
|
|
|
83 |
dir_zip='/tmp/works/preprocess/nasilhong_f_v1_front/crop_video_front_one_piece_dress_nodded_cut.zip'
|
84 |
dir_target='/tmp/works/preprocess/nasilhong_f_v1_front/'
|
|
|
|
|
|
|
|
|
85 |
import zipfile; zipfile.ZipFile(dir_zip, 'r').extractall(dir_target)
|
|
|
86 |
|
87 |
self.config_path = os.path.join(stf_path, config_path)
|
88 |
self.checkpoint_path = os.path.join(stf_path, checkpoint_path)
|
|
|
80 |
):
|
81 |
#os.makedirs(root_path, exist_ok=True)
|
82 |
import shutil; shutil.copytree('/home/user/app/stf/works', '/tmp/works', dirs_exist_ok=True)
|
83 |
+
|
84 |
+
import zipfile
|
85 |
dir_zip='/tmp/works/preprocess/nasilhong_f_v1_front/crop_video_front_one_piece_dress_nodded_cut.zip'
|
86 |
dir_target='/tmp/works/preprocess/nasilhong_f_v1_front/'
|
87 |
+
zipfile.ZipFile(dir_zip, 'r').extractall(dir_target)
|
88 |
+
|
89 |
+
dir_zip='/tmp/works/preprocess/nasilhong_f_v1_front/front_one_piece_dress_nodded_cut.zip'
|
90 |
+
dir_target='/tmp/works/preprocess/nasilhong_f_v1_front/'
|
91 |
import zipfile; zipfile.ZipFile(dir_zip, 'r').extractall(dir_target)
|
92 |
+
|
93 |
|
94 |
self.config_path = os.path.join(stf_path, config_path)
|
95 |
self.checkpoint_path = os.path.join(stf_path, checkpoint_path)
|