jhansss commited on
Commit
20bd7b4
·
1 Parent(s): 9d28841

cli.py use Path for file paths

Browse files
Files changed (1) hide show
  1. cli.py +8 -4
cli.py CHANGED
@@ -1,5 +1,6 @@
1
  from argparse import ArgumentParser
2
  from logging import getLogger
 
3
 
4
  import soundfile as sf
5
  import yaml
@@ -12,13 +13,15 @@ logger = getLogger(__name__)
12
 
13
  def get_parser():
14
  parser = ArgumentParser()
15
- parser.add_argument("--query_audio", type=str, required=True)
16
- parser.add_argument("--config_path", type=str, default="config/cli/yaoyin_default.yaml")
17
- parser.add_argument("--output_audio", type=str, required=True)
 
 
18
  return parser
19
 
20
 
21
- def load_config(config_path: str):
22
  with open(config_path, "r") as f:
23
  config = yaml.safe_load(f)
24
  return config
@@ -39,6 +42,7 @@ def main():
39
  f"Input: {args.query_audio}, Output: {args.output_audio}, ASR results: {results['asr_text']}, LLM results: {results['llm_text']}"
40
  )
41
  svs_audio, svs_sample_rate = results["svs_audio"]
 
42
  sf.write(args.output_audio, svs_audio, svs_sample_rate)
43
 
44
 
 
1
  from argparse import ArgumentParser
2
  from logging import getLogger
3
+ from pathlib import Path
4
 
5
  import soundfile as sf
6
  import yaml
 
13
 
14
  def get_parser():
15
  parser = ArgumentParser()
16
+ parser.add_argument("--query_audio", type=Path, required=True)
17
+ parser.add_argument(
18
+ "--config_path", type=Path, default="config/cli/yaoyin_default.yaml"
19
+ )
20
+ parser.add_argument("--output_audio", type=Path, required=True)
21
  return parser
22
 
23
 
24
+ def load_config(config_path: Path):
25
  with open(config_path, "r") as f:
26
  config = yaml.safe_load(f)
27
  return config
 
42
  f"Input: {args.query_audio}, Output: {args.output_audio}, ASR results: {results['asr_text']}, LLM results: {results['llm_text']}"
43
  )
44
  svs_audio, svs_sample_rate = results["svs_audio"]
45
+ args.output_audio.parent.mkdir(parents=True, exist_ok=True)
46
  sf.write(args.output_audio, svs_audio, svs_sample_rate)
47
 
48