EvanTHU commited on
Commit
d524b42
·
verified ·
1 Parent(s): 9d8640c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
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/gen"
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/gen/'
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/gen/'
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/gen/'
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/gen/'
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)