Update app_parallel.py

app_parallel.py

@@ -29,7 +29,7 @@ from concurrent.futures import ProcessPoolExecutor, as_completed, ThreadPoolExec
 
 # Initialize ProcessPoolExecutor for parallel processing
 executor = ThreadPoolExecutor(max_workers=3)
-
+torch.cuda.empty_cache()
 
 class AnimationConfig:
     def __init__(self, driven_audio_path, source_image_path, result_folder,pose_style,expression_scale,enhancer,still,preprocess,ref_pose_video_path, image_hardcoded):
@@ -126,7 +126,8 @@ def process_chunk(audio_chunk, preprocessed_data, args):
     # Further processing with animate_from_coeff using the coeff_path
     animate_from_coeff = AnimateFromCoeff(free_view_checkpoint, mapping_checkpoint, 
                                             facerender_yaml_path, args.device)
-    
+
+    torch.cuda.empty_cache()
     data = get_facerender_data(coeff_path, crop_pic_path, first_coeff_path, audio_chunk, 
                                 args.batch_size, args.input_yaw, args.input_pitch, args.input_roll, 
                                 expression_scale=args.expression_scale, still_mode=args.still, preprocess=args.preprocess)