Update app.py

app.py

@@ -26,7 +26,7 @@ from PIL import Image
 start_time = time.time()
 
 class AnimationConfig:
-    def __init__(self, driven_audio_path, source_image_path, result_folder,pose_style,expression_scale,still,preprocess,ref_pose_video_path):
+    def __init__(self, driven_audio_path, source_image_path, result_folder,pose_style,expression_scale,enhancer,still,preprocess,ref_pose_video_path):
         self.driven_audio = driven_audio_path
         self.source_image = source_image_path
         self.ref_eyeblink = None
@@ -39,7 +39,7 @@ class AnimationConfig:
         self.input_yaw = None
         self.input_pitch = None
         self.input_roll = None
-        self.enhancer = None
+        self.enhancer = enhancer
         self.background_enhancer = None
         self.cpu = False
         self.face3dvis = False
@@ -341,7 +341,7 @@ def generate_video():
         return "An error occurred", 500
     
     # Example of using the class with some hypothetical paths
-    args = AnimationConfig(driven_audio_path=driven_audio_path, source_image_path=source_image_path, result_folder=result_folder, pose_style=pose_style, expression_scale=expression_scale,still=still,preprocess=preprocess,ref_pose_video_path=ref_pose_video_path)
+    args = AnimationConfig(driven_audio_path=driven_audio_path, source_image_path=source_image_path, result_folder=result_folder, pose_style=pose_style, expression_scale=expression_scale,enhancer=enhancer,still=still,preprocess=preprocess,ref_pose_video_path=ref_pose_video_path)
         
     if torch.cuda.is_available() and not args.cpu:
         args.device = "cuda"