Spanicin commited on
Commit
a648386
1 Parent(s): 7dac13b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -25,7 +25,7 @@ import time
25
  start_time = time.time()
26
 
27
  class AnimationConfig:
28
- def __init__(self, driven_audio_path, source_image_path, result_folder,pose_style,expression_scale,enhancer,still,preprocess,ref_pose_video_path):
29
  self.driven_audio = driven_audio_path
30
  self.source_image = source_image_path
31
  self.ref_eyeblink = None
@@ -38,7 +38,7 @@ class AnimationConfig:
38
  self.input_yaw = None
39
  self.input_pitch = None
40
  self.input_roll = None
41
- self.enhancer = enhancer
42
  self.background_enhancer = None
43
  self.cpu = False
44
  self.face3dvis = False
@@ -324,7 +324,7 @@ def generate_video():
324
  return "An error occurred", 500
325
 
326
  # Example of using the class with some hypothetical paths
327
- 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)
328
 
329
  if torch.cuda.is_available() and not args.cpu:
330
  args.device = "cuda"
 
25
  start_time = time.time()
26
 
27
  class AnimationConfig:
28
+ def __init__(self, driven_audio_path, source_image_path, result_folder,pose_style,expression_scale,still,preprocess,ref_pose_video_path):
29
  self.driven_audio = driven_audio_path
30
  self.source_image = source_image_path
31
  self.ref_eyeblink = None
 
38
  self.input_yaw = None
39
  self.input_pitch = None
40
  self.input_roll = None
41
+ self.enhancer = None
42
  self.background_enhancer = None
43
  self.cpu = False
44
  self.face3dvis = False
 
324
  return "An error occurred", 500
325
 
326
  # Example of using the class with some hypothetical paths
327
+ 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)
328
 
329
  if torch.cuda.is_available() and not args.cpu:
330
  args.device = "cuda"