zyingt commited on
Commit
f1eca78
1 Parent(s): fb23346

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +4 -2
inference.py CHANGED
@@ -141,7 +141,7 @@ def build_parser():
141
 
142
  def main(args_list):
143
  # Parse arguments and config
144
- args = parser.parse_args(args_list)
145
  cfg = load_config(args.config)
146
 
147
  # CUDA settings
@@ -151,7 +151,9 @@ def main(args_list):
151
  inferencer = build_inference(args, cfg)
152
 
153
  # Run inference
154
- inferencer.inference()
 
 
155
 
156
 
157
  if __name__ == "__main__":
 
141
 
142
  def main(args_list):
143
  # Parse arguments and config
144
+ args = build_parser().parse_args(args_list)
145
  cfg = load_config(args.config)
146
 
147
  # CUDA settings
 
151
  inferencer = build_inference(args, cfg)
152
 
153
  # Run inference
154
+ output_audio = inferencer.inference()
155
+
156
+ return output_audio
157
 
158
 
159
  if __name__ == "__main__":