Update app_parallel.py

app_parallel.py

@@ -115,7 +115,7 @@ def process_chunk(audio_chunk, preprocessed_data, args):
     print("first_coeff_path",first_coeff_path)
     print("crop_pic_path",crop_pic_path)
     print("crop_info",crop_info)
-
+    torch.cuda.empty_cache()
     batch = get_data(first_coeff_path, audio_chunk, args.device, ref_eyeblink_coeff_path=None, still=args.still)
     audio_to_coeff = Audio2Coeff(audio2pose_checkpoint, audio2pose_yaml_path, 
                                 audio2exp_checkpoint, audio2exp_yaml_path, 
@@ -129,7 +129,7 @@ def process_chunk(audio_chunk, preprocessed_data, args):
     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)
-
+    torch.cuda.empty_cache()
     print("Will Enter Animation")
     result, base64_video, temp_file_path, _ = animate_from_coeff.generate(data, args.result_dir, args.source_image, crop_info, 
                                     enhancer=args.enhancer, background_enhancer=args.background_enhancer, preprocess=args.preprocess)
@@ -140,7 +140,7 @@ def process_chunk(audio_chunk, preprocessed_data, args):
     app.config['temp_response'] = base64_video
     app.config['final_video_path'] = temp_file_path
     # app.config['final_video_duration'] = duration
-    
+    torch.cuda.empty_cache()
     return base64_video, temp_file_path