EvanTHU commited on
Commit
ce8bed7
·
verified ·
1 Parent(s): e954aa9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -58,7 +58,7 @@ opt_path = hf_hub_download(
58
 
59
 
60
  os.makedirs("tmp", exist_ok=True)
61
- os.environ['GRADIO_TEMP_DIR'] = './tmp'
62
 
63
  def generate_md5(input_string):
64
  # Encode the string and compute the MD5 hash
@@ -85,7 +85,7 @@ HEAD = """<div class="embed_hidden">
85
  """
86
 
87
  edit_config = yaml_to_box('options/edit.yaml')
88
- os.environ['GRADIO_TEMP_DIR'] = './tmp'
89
  CSS = """
90
  .retrieved_video {
91
  position: relative;
@@ -119,7 +119,7 @@ def generate_video_from_text(text, opt, pipeline):
119
  texts = [text]
120
  motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
121
 
122
- save_dir = './tmp/gen/'
123
  filename = generate_md5(str(time.time())) + ".mp4"
124
  save_path = pjoin(save_dir, str(filename))
125
  os.makedirs(save_dir, exist_ok=True)
@@ -201,7 +201,7 @@ def reweighting(text, idx, weight, opt, pipeline):
201
  texts = [text, text]
202
  motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
203
 
204
- save_dir = './tmp/gen/'
205
  filenames = [generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4"]
206
  save_paths = [pjoin(save_dir, str(filenames[0])), pjoin(save_dir, str(filenames[1]))]
207
  os.makedirs(save_dir, exist_ok=True)
@@ -280,7 +280,7 @@ def generate_example_based_motion(text, chunk_size, example_based_steps_end, tem
280
  texts = [text for _ in range(num_motion)]
281
  motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
282
 
283
- save_dir = './tmp/gen/'
284
  filenames = [generate_md5(str(time.time())) + ".mp4" for _ in range(num_motion)]
285
  save_paths = [pjoin(save_dir, str(filenames[i])) for i in range(num_motion)]
286
  os.makedirs(save_dir, exist_ok=True)
@@ -365,7 +365,7 @@ def transfer_style(text, style_text, style_transfer_steps_end, opt, pipeline):
365
  texts = [style_text, text, text]
366
  motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
367
 
368
- save_dir = './tmp/gen/'
369
  filenames = [generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4"]
370
  save_paths = [pjoin(save_dir, str(filenames[0])), pjoin(save_dir, str(filenames[1])), pjoin(save_dir, str(filenames[2]))]
371
  os.makedirs(save_dir, exist_ok=True)
 
58
 
59
 
60
  os.makedirs("tmp", exist_ok=True)
61
+ os.environ['GRADIO_TEMP_DIR'] = 'tmp'
62
 
63
  def generate_md5(input_string):
64
  # Encode the string and compute the MD5 hash
 
85
  """
86
 
87
  edit_config = yaml_to_box('options/edit.yaml')
88
+ os.environ['GRADIO_TEMP_DIR'] = 'tmp'
89
  CSS = """
90
  .retrieved_video {
91
  position: relative;
 
119
  texts = [text]
120
  motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
121
 
122
+ save_dir = 'tmp/gen/'
123
  filename = generate_md5(str(time.time())) + ".mp4"
124
  save_path = pjoin(save_dir, str(filename))
125
  os.makedirs(save_dir, exist_ok=True)
 
201
  texts = [text, text]
202
  motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
203
 
204
+ save_dir = 'tmp/gen/'
205
  filenames = [generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4"]
206
  save_paths = [pjoin(save_dir, str(filenames[0])), pjoin(save_dir, str(filenames[1]))]
207
  os.makedirs(save_dir, exist_ok=True)
 
280
  texts = [text for _ in range(num_motion)]
281
  motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
282
 
283
+ save_dir = 'tmp/gen/'
284
  filenames = [generate_md5(str(time.time())) + ".mp4" for _ in range(num_motion)]
285
  save_paths = [pjoin(save_dir, str(filenames[i])) for i in range(num_motion)]
286
  os.makedirs(save_dir, exist_ok=True)
 
365
  texts = [style_text, text, text]
366
  motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
367
 
368
+ save_dir = 'tmp/gen/'
369
  filenames = [generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4"]
370
  save_paths = [pjoin(save_dir, str(filenames[0])), pjoin(save_dir, str(filenames[1])), pjoin(save_dir, str(filenames[2]))]
371
  os.makedirs(save_dir, exist_ok=True)