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