Update app.py

app.py

@@ -117,7 +117,7 @@ def process_chunk(audio_chunk, args):
     # with open(crop_info_path , "rb") as f:
     #         crop_info = json.load(f)
 
-    first_frame_dir = os.path.join(save_dir, 'first_frame_dir')
+    first_frame_dir = os.path.join(args.result_dir, 'first_frame_dir')
     os.makedirs(first_frame_dir, exist_ok=True)
     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)