Spaces:
Running
on
Zero
Running
on
Zero
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)
|