Spaces:
Running
Running
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__":
|