Update app.py
Browse files
app.py
CHANGED
@@ -25,8 +25,7 @@ ckptdir = './checkpoints/t2m/release'
|
|
25 |
os.makedirs(ckptdir, exist_ok=True)
|
26 |
|
27 |
|
28 |
-
os.environ['GRADIO_TEMP_DIR']="temp
|
29 |
-
os.environ['COMMANDLINE_ARGS'] = '--no-gradio-queue'
|
30 |
|
31 |
mean_path = hf_hub_download(
|
32 |
repo_id="EvanTHU/MotionCLR",
|
@@ -148,7 +147,7 @@ def generate_video_from_text(text, opt, pipeline):
|
|
148 |
texts = [text]
|
149 |
motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
|
150 |
|
151 |
-
save_dir = 'temp/
|
152 |
filename = generate_md5(str(time.time())) + ".mp4"
|
153 |
save_path = pjoin(save_dir, str(filename))
|
154 |
os.makedirs(save_dir, exist_ok=True)
|
@@ -230,7 +229,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 = '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,7 +308,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 = '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,7 +393,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 = '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)
|
|
|
25 |
os.makedirs(ckptdir, exist_ok=True)
|
26 |
|
27 |
|
28 |
+
os.environ['GRADIO_TEMP_DIR']="temp"
|
|
|
29 |
|
30 |
mean_path = hf_hub_download(
|
31 |
repo_id="EvanTHU/MotionCLR",
|
|
|
147 |
texts = [text]
|
148 |
motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
|
149 |
|
150 |
+
save_dir = 'temp/'
|
151 |
filename = generate_md5(str(time.time())) + ".mp4"
|
152 |
save_path = pjoin(save_dir, str(filename))
|
153 |
os.makedirs(save_dir, exist_ok=True)
|
|
|
229 |
texts = [text, text]
|
230 |
motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
|
231 |
|
232 |
+
save_dir = 'temp/'
|
233 |
filenames = [generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4"]
|
234 |
save_paths = [pjoin(save_dir, str(filenames[0])), pjoin(save_dir, str(filenames[1]))]
|
235 |
os.makedirs(save_dir, exist_ok=True)
|
|
|
308 |
texts = [text for _ in range(num_motion)]
|
309 |
motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
|
310 |
|
311 |
+
save_dir = 'temp/'
|
312 |
filenames = [generate_md5(str(time.time())) + ".mp4" for _ in range(num_motion)]
|
313 |
save_paths = [pjoin(save_dir, str(filenames[i])) for i in range(num_motion)]
|
314 |
os.makedirs(save_dir, exist_ok=True)
|
|
|
393 |
texts = [style_text, text, text]
|
394 |
motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
|
395 |
|
396 |
+
save_dir = 'temp/'
|
397 |
filenames = [generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4"]
|
398 |
save_paths = [pjoin(save_dir, str(filenames[0])), pjoin(save_dir, str(filenames[1])), pjoin(save_dir, str(filenames[2]))]
|
399 |
os.makedirs(save_dir, exist_ok=True)
|