Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -25,8 +25,8 @@ ckptdir = './checkpoints/t2m/release'
|
|
25 |
os.makedirs(ckptdir, exist_ok=True)
|
26 |
|
27 |
|
28 |
-
os.environ['GRADIO_TEMP_DIR']="
|
29 |
-
os.environ['GRADIO_ALLOWED_PATHS']="
|
30 |
|
31 |
mean_path = hf_hub_download(
|
32 |
repo_id="EvanTHU/MotionCLR",
|
@@ -58,7 +58,7 @@ opt_path = hf_hub_download(
|
|
58 |
|
59 |
|
60 |
|
61 |
-
os.makedirs("
|
62 |
|
63 |
def generate_md5(input_string):
|
64 |
# Encode the string and compute the MD5 hash
|
@@ -146,7 +146,7 @@ def generate_video_from_text(text, opt, pipeline):
|
|
146 |
texts = [text]
|
147 |
motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
|
148 |
|
149 |
-
save_dir = '
|
150 |
filename = generate_md5(str(time.time())) + ".mp4"
|
151 |
save_path = pjoin(save_dir, str(filename))
|
152 |
os.makedirs(save_dir, exist_ok=True)
|
@@ -230,7 +230,7 @@ def reweighting(text, idx, weight, opt, pipeline):
|
|
230 |
texts = [text, text]
|
231 |
motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
|
232 |
|
233 |
-
save_dir = '
|
234 |
filenames = [generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4"]
|
235 |
save_paths = [pjoin(save_dir, str(filenames[0])), pjoin(save_dir, str(filenames[1]))]
|
236 |
os.makedirs(save_dir, exist_ok=True)
|
@@ -309,7 +309,7 @@ def generate_example_based_motion(text, chunk_size, example_based_steps_end, tem
|
|
309 |
texts = [text for _ in range(num_motion)]
|
310 |
motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
|
311 |
|
312 |
-
save_dir = '
|
313 |
filenames = [generate_md5(str(time.time())) + ".mp4" for _ in range(num_motion)]
|
314 |
save_paths = [pjoin(save_dir, str(filenames[i])) for i in range(num_motion)]
|
315 |
os.makedirs(save_dir, exist_ok=True)
|
@@ -394,7 +394,7 @@ def transfer_style(text, style_text, style_transfer_steps_end, opt, pipeline):
|
|
394 |
texts = [style_text, text, text]
|
395 |
motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
|
396 |
|
397 |
-
save_dir = '
|
398 |
filenames = [generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4"]
|
399 |
save_paths = [pjoin(save_dir, str(filenames[0])), pjoin(save_dir, str(filenames[1])), pjoin(save_dir, str(filenames[2]))]
|
400 |
os.makedirs(save_dir, exist_ok=True)
|
|
|
25 |
os.makedirs(ckptdir, exist_ok=True)
|
26 |
|
27 |
|
28 |
+
os.environ['GRADIO_TEMP_DIR']="temp"
|
29 |
+
os.environ['GRADIO_ALLOWED_PATHS']="temp"
|
30 |
|
31 |
mean_path = hf_hub_download(
|
32 |
repo_id="EvanTHU/MotionCLR",
|
|
|
58 |
|
59 |
|
60 |
|
61 |
+
os.makedirs("temp", exist_ok=True)
|
62 |
|
63 |
def generate_md5(input_string):
|
64 |
# Encode the string and compute the MD5 hash
|
|
|
146 |
texts = [text]
|
147 |
motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
|
148 |
|
149 |
+
save_dir = 'temp/'
|
150 |
filename = generate_md5(str(time.time())) + ".mp4"
|
151 |
save_path = pjoin(save_dir, str(filename))
|
152 |
os.makedirs(save_dir, exist_ok=True)
|
|
|
230 |
texts = [text, text]
|
231 |
motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
|
232 |
|
233 |
+
save_dir = 'temp/'
|
234 |
filenames = [generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4"]
|
235 |
save_paths = [pjoin(save_dir, str(filenames[0])), pjoin(save_dir, str(filenames[1]))]
|
236 |
os.makedirs(save_dir, exist_ok=True)
|
|
|
309 |
texts = [text for _ in range(num_motion)]
|
310 |
motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
|
311 |
|
312 |
+
save_dir = 'temp/'
|
313 |
filenames = [generate_md5(str(time.time())) + ".mp4" for _ in range(num_motion)]
|
314 |
save_paths = [pjoin(save_dir, str(filenames[i])) for i in range(num_motion)]
|
315 |
os.makedirs(save_dir, exist_ok=True)
|
|
|
394 |
texts = [style_text, text, text]
|
395 |
motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
|
396 |
|
397 |
+
save_dir = 'temp/'
|
398 |
filenames = [generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4"]
|
399 |
save_paths = [pjoin(save_dir, str(filenames[0])), pjoin(save_dir, str(filenames[1])), pjoin(save_dir, str(filenames[2]))]
|
400 |
os.makedirs(save_dir, exist_ok=True)
|