Update app_parallel.py

app_parallel.py

@@ -109,9 +109,9 @@ def process_chunk(audio_chunk, preprocessed_data, args):
 
     first_coeff_path = preprocessed_data["first_coeff_path"]
     crop_pic_path = preprocessed_data["crop_pic_path"]
-    crop_info_path = preprocessed_data["crop_info_path"]
+    crop_info_path = "/home/user/app/preprocess_data/crop_info.json"
     with open(crop_info_path , "rb") as f:
-            crop_info = pickle.load(f)
+            crop_info = json.load(f)
 
     print(f"Loaded existing preprocessed data")
     print("first_coeff_path",first_coeff_path)
@@ -224,7 +224,7 @@ def run_preprocessing(args):
     global path_of_lm_croper, path_of_net_recon_model, dir_of_BFM_fitting
     first_frame_dir = os.path.join(args.result_dir, 'first_frame_dir')
     os.makedirs(first_frame_dir, exist_ok=True)
-    fixed_temp_dir = "/tmp/preprocess_data"
+    fixed_temp_dir = "/home/user/app/preprocess_data/"
     os.makedirs(fixed_temp_dir, exist_ok=True)
     preprocessed_data_path = os.path.join(fixed_temp_dir, "preprocessed_data.pkl")
 
@@ -233,30 +233,8 @@ def run_preprocessing(args):
         with open(preprocessed_data_path, "rb") as f:
             preprocessed_data = pickle.load(f)
         print("Loaded existing preprocessed data from:", preprocessed_data_path)
-    else:
-        print("Running preprocessing...")
-        preprocess_model = CropAndExtract(path_of_lm_croper, path_of_net_recon_model, dir_of_BFM_fitting, args.device)
-        first_coeff_path, crop_pic_path, crop_info = preprocess_model.generate(args.source_image, first_frame_dir, args.preprocess, source_image_flag=True)
-        first_coeff_new_path = os.path.join(fixed_temp_dir, os.path.basename(first_coeff_path))
-        crop_pic_new_path = os.path.join(fixed_temp_dir, os.path.basename(crop_pic_path))
-        crop_info_new_path = os.path.join(fixed_temp_dir, "crop_info.pkl")
-        shutil.move(first_coeff_path, first_coeff_new_path)
-        shutil.move(crop_pic_path, crop_pic_new_path)
-
-        with open(crop_info_new_path, "wb") as f:
-            pickle.dump(crop_info, f)
-
-        preprocessed_data = {"first_coeff_path": first_coeff_new_path,
-                            "crop_pic_path": crop_pic_new_path,
-                            "crop_info_path": crop_info_new_path}
-
-
-        with open(preprocessed_data_path, "wb") as f:
-            pickle.dump(preprocessed_data, f)
-        print(f"Preprocessed data saved to: {preprocessed_data_path}")
 
     return preprocessed_data
-
     
 client = OpenAI(api_key="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")