Spaces:
Runtime error
Runtime error
Update inference.py
Browse files- 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 =
|
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__":
|