Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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,
|
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 =
|
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,
|
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"
|