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)
|