EvanTHU commited on
Commit
fdf8b2b
·
verified ·
1 Parent(s): 4ee8c0d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
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']="/home/user/app/temp"
29
- os.environ['GRADIO_ALLOWED_PATHS']="/home/user/app/temp"
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("/home/user/app/temp", exist_ok=True)
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 = '/home/user/app/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,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 = '/home/user/app/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 +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 = '/home/user/app/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 +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 = '/home/user/app/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
+ 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)