Update app.py

app.py

@@ -25,7 +25,7 @@ ckptdir = './checkpoints/t2m/release'
 os.makedirs(ckptdir, exist_ok=True)
 
 
-os.environ['GRADIO_TEMP_DIR']="temp"
+os.environ['GRADIO_TEMP_DIR']="/home/user/app/temp"
 
 mean_path = hf_hub_download(
     repo_id="EvanTHU/MotionCLR",
@@ -57,8 +57,8 @@ opt_path = hf_hub_download(
 
 
 
-os.makedirs("temp", exist_ok=True)
-os.environ['GRADIO_TEMP_DIR'] = 'temp'
+os.makedirs("/home/user/app/temp", exist_ok=True)
+os.environ['GRADIO_TEMP_DIR'] = '/home/user/app/temp'
 
 def generate_md5(input_string):
     # Encode the string and compute the MD5 hash
@@ -113,7 +113,7 @@ HEAD = ("""<div>
 
 
 edit_config = yaml_to_box('options/edit.yaml')
-os.environ['GRADIO_TEMP_DIR'] = 'temp'
+os.environ['GRADIO_TEMP_DIR'] = '/home/user/app/temp'
 CSS = """
 .retrieved_video {
     position: relative;
@@ -147,7 +147,7 @@ def generate_video_from_text(text, opt, pipeline):
     texts = [text]
     motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
     
-    save_dir = 'temp/'
+    save_dir = '/home/user/app/temp/'
     filename = generate_md5(str(time.time())) + ".mp4"
     save_path = pjoin(save_dir, str(filename))
     os.makedirs(save_dir, exist_ok=True)
@@ -231,7 +231,7 @@ def reweighting(text, idx, weight, opt, pipeline):
     texts = [text, text]
     motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
     
-    save_dir = 'temp/'
+    save_dir = '/home/user/app/temp/'
     filenames = [generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4"]
     save_paths = [pjoin(save_dir, str(filenames[0])), pjoin(save_dir, str(filenames[1]))]
     os.makedirs(save_dir, exist_ok=True)
@@ -310,7 +310,7 @@ def generate_example_based_motion(text, chunk_size, example_based_steps_end, tem
     texts = [text for _ in range(num_motion)]
     motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
     
-    save_dir = 'temp/'
+    save_dir = '/home/user/app/temp/'
     filenames = [generate_md5(str(time.time())) + ".mp4" for _ in range(num_motion)]
     save_paths = [pjoin(save_dir, str(filenames[i])) for i in range(num_motion)]
     os.makedirs(save_dir, exist_ok=True)
@@ -395,7 +395,7 @@ def transfer_style(text, style_text, style_transfer_steps_end, opt, pipeline):
     texts = [style_text, text, text]
     motion_lens = [opt.motion_length * opt.fps for _ in range(opt.num_samples)]
     
-    save_dir = 'temp/'
+    save_dir = '/home/user/app/temp/'
     filenames = [generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4", generate_md5(str(time.time())) + ".mp4"]
     save_paths = [pjoin(save_dir, str(filenames[0])), pjoin(save_dir, str(filenames[1])), pjoin(save_dir, str(filenames[2]))]
     os.makedirs(save_dir, exist_ok=True)