Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -58,7 +58,7 @@ opt_path = hf_hub_download(
|
|
58 |
|
59 |
|
60 |
os.makedirs("tmp", exist_ok=True)
|
61 |
-
os.environ['GRADIO_TEMP_DIR'] = '
|
62 |
|
63 |
def generate_md5(input_string):
|
64 |
# Encode the string and compute the MD5 hash
|
@@ -85,7 +85,7 @@ HEAD = """<div class="embed_hidden">
|
|
85 |
"""
|
86 |
|
87 |
edit_config = yaml_to_box('options/edit.yaml')
|
88 |
-
os.environ['GRADIO_TEMP_DIR'] = '
|
89 |
CSS = """
|
90 |
.retrieved_video {
|
91 |
position: relative;
|
@@ -119,7 +119,7 @@ def generate_video_from_text(text, opt, pipeline):
|
|
119 |
texts = [text]
|
120 |
motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
|
121 |
|
122 |
-
save_dir = '
|
123 |
filename = generate_md5(str(time.time())) + ".mp4"
|
124 |
save_path = pjoin(save_dir, str(filename))
|
125 |
os.makedirs(save_dir, exist_ok=True)
|
@@ -201,7 +201,7 @@ def reweighting(text, idx, weight, opt, pipeline):
|
|
201 |
texts = [text, text]
|
202 |
motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
|
203 |
|
204 |
-
save_dir = '
|
205 |
filenames = [generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4"]
|
206 |
save_paths = [pjoin(save_dir, str(filenames[0])), pjoin(save_dir, str(filenames[1]))]
|
207 |
os.makedirs(save_dir, exist_ok=True)
|
@@ -280,7 +280,7 @@ def generate_example_based_motion(text, chunk_size, example_based_steps_end, tem
|
|
280 |
texts = [text for _ in range(num_motion)]
|
281 |
motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
|
282 |
|
283 |
-
save_dir = '
|
284 |
filenames = [generate_md5(str(time.time())) + ".mp4" for _ in range(num_motion)]
|
285 |
save_paths = [pjoin(save_dir, str(filenames[i])) for i in range(num_motion)]
|
286 |
os.makedirs(save_dir, exist_ok=True)
|
@@ -365,7 +365,7 @@ def transfer_style(text, style_text, style_transfer_steps_end, opt, pipeline):
|
|
365 |
texts = [style_text, text, text]
|
366 |
motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
|
367 |
|
368 |
-
save_dir = '
|
369 |
filenames = [generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4"]
|
370 |
save_paths = [pjoin(save_dir, str(filenames[0])), pjoin(save_dir, str(filenames[1])), pjoin(save_dir, str(filenames[2]))]
|
371 |
os.makedirs(save_dir, exist_ok=True)
|
|
|
58 |
|
59 |
|
60 |
os.makedirs("tmp", exist_ok=True)
|
61 |
+
os.environ['GRADIO_TEMP_DIR'] = 'tmp'
|
62 |
|
63 |
def generate_md5(input_string):
|
64 |
# Encode the string and compute the MD5 hash
|
|
|
85 |
"""
|
86 |
|
87 |
edit_config = yaml_to_box('options/edit.yaml')
|
88 |
+
os.environ['GRADIO_TEMP_DIR'] = 'tmp'
|
89 |
CSS = """
|
90 |
.retrieved_video {
|
91 |
position: relative;
|
|
|
119 |
texts = [text]
|
120 |
motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
|
121 |
|
122 |
+
save_dir = 'tmp/gen/'
|
123 |
filename = generate_md5(str(time.time())) + ".mp4"
|
124 |
save_path = pjoin(save_dir, str(filename))
|
125 |
os.makedirs(save_dir, exist_ok=True)
|
|
|
201 |
texts = [text, text]
|
202 |
motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
|
203 |
|
204 |
+
save_dir = 'tmp/gen/'
|
205 |
filenames = [generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4"]
|
206 |
save_paths = [pjoin(save_dir, str(filenames[0])), pjoin(save_dir, str(filenames[1]))]
|
207 |
os.makedirs(save_dir, exist_ok=True)
|
|
|
280 |
texts = [text for _ in range(num_motion)]
|
281 |
motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
|
282 |
|
283 |
+
save_dir = 'tmp/gen/'
|
284 |
filenames = [generate_md5(str(time.time())) + ".mp4" for _ in range(num_motion)]
|
285 |
save_paths = [pjoin(save_dir, str(filenames[i])) for i in range(num_motion)]
|
286 |
os.makedirs(save_dir, exist_ok=True)
|
|
|
365 |
texts = [style_text, text, text]
|
366 |
motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
|
367 |
|
368 |
+
save_dir = 'tmp/gen/'
|
369 |
filenames = [generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4"]
|
370 |
save_paths = [pjoin(save_dir, str(filenames[0])), pjoin(save_dir, str(filenames[1])), pjoin(save_dir, str(filenames[2]))]
|
371 |
os.makedirs(save_dir, exist_ok=True)
|