Update app.py

app.py

@@ -142,9 +142,10 @@ def main(args):
     # first_coeff_path, crop_pic_path, crop_info =  preprocess_model.generate(pic_path, first_frame_dir, args.preprocess,\
                                                                             #  source_image_flag=True, pic_size=args.size)
 
-    preprocess_dir = os.path.join(TEMP_DIR.name, "preprocess_data")
-    os.makedirs(preprocess_dir, exist_ok=True)
-    preprocessed_data_path = os.path.join(preprocess_dir, "preprocessed_data.pkl")
+
+    fixed_temp_dir = "/tmp/preprocess_data"
+    os.makedirs(fixed_temp_dir, exist_ok=True)
+    preprocessed_data_path = os.path.join(fixed_temp_dir, "preprocessed_data.pkl")
 
     if os.path.exists(preprocessed_data_path) and image_hardcoded == "yes":
         print("Loading preprocessed data...")
@@ -547,7 +548,7 @@ def check_generation_status():
                 print("Deleted video file:", final_video_path)
 
             # TEMP_DIR.cleanup()
-            preprocess_dir = os.path.join(TEMP_DIR.name, "preprocess_data")
+            preprocess_dir = os.path.join("/tmp", "preprocess_data")
             custom_cleanup(TEMP_DIR.name, preprocess_dir)
 
             print("Temporary files cleaned up, but preprocess_data is retained.")