Update app.py

app.py

@@ -151,9 +151,12 @@ def main(args):
         print("Loading preprocessed data...")
         with open(preprocessed_data_path, "rb") as f:
             preprocessed_data = pickle.load(f)
-        first_coeff_path = preprocessed_data["first_coeff_path"]
-        crop_pic_path = preprocessed_data["crop_pic_path"]
-        crop_info = preprocessed_data["crop_info"]
+        first_coeff_new_path = preprocessed_data["first_coeff_path"]
+        crop_pic_new_path = preprocessed_data["crop_pic_path"]
+        crop_info_path = preprocessed_data["crop_info_path"]
+        with open(crop_info_path, "rb") as f:
+            crop_info = pickle.load(f)
+            
         print(f"Loaded existing preprocessed data from: {preprocessed_data_path}")
     
     else:
@@ -161,13 +164,17 @@ 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)
         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, os.path.basename(crop_info))
+        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)
-        shutil.move(crop_info, crop_info_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": crop_info_new_path}
+                            "crop_info_path": crop_info_new_path}
+
 
         with open(preprocessed_data_path, "wb") as f:
             pickle.dump(preprocessed_data, f)
@@ -175,7 +182,7 @@ def main(args):
     
     print('first_coeff_path ',first_coeff_new_path)
     print('crop_pic_path ',crop_pic_new_path)
-    print('crop_info ',crop_info_new_path)
+    print('crop_info ',crop_info)
 
     if first_coeff_path is None:
         print("Can't get the coeffs of the input")