Spaces:
Running
on
A10G
Running
on
A10G
Update modules/sadtalker_test.py
Browse files
modules/sadtalker_test.py
CHANGED
@@ -53,7 +53,7 @@ class SadTalker():
|
|
53 |
facerender_yaml_path, device)
|
54 |
self.device = device
|
55 |
|
56 |
-
def test(self, source_image, driven_audio, still_mode, use_enhancer, result_dir):
|
57 |
|
58 |
time_tag = strftime("%Y_%m_%d_%H.%M.%S")
|
59 |
save_dir = os.path.join(result_dir, time_tag)
|
|
|
53 |
facerender_yaml_path, device)
|
54 |
self.device = device
|
55 |
|
56 |
+
def test(self, source_image, driven_audio, still_mode, use_enhancer, result_dir='./'):
|
57 |
|
58 |
time_tag = strftime("%Y_%m_%d_%H.%M.%S")
|
59 |
save_dir = os.path.join(result_dir, time_tag)
|