vinthony commited on
Commit
e4c1fff
1 Parent(s): bb90c5d

Update modules/sadtalker_test.py

Browse files
Files changed (1) hide show
  1. modules/sadtalker_test.py +1 -1
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)