chore(format): run black on main (#501)

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
github-actions[bot]
2024-06-29 17:32:49 +09:00
committed by GitHub
co-authored by github-actions[bot]
parent c98fd169db
commit 7db7d08b5a
2 changed files with 34 additions and 11 deletions
+15 -6
View File
@@ -35,16 +35,19 @@ def main(texts: List[str], spk: Optional[str] = None):
else:
logger.error("Models load failed.")
sys.exit(1)
if spk is None:
spk = chat.sample_random_speaker()
logger.info("Use speaker:")
print(spk)
logger.info("Start inference.")
wavs = chat.infer(texts, params_infer_code=ChatTTS.Chat.InferCodeParams(
spk_emb=spk,
))
wavs = chat.infer(
texts,
params_infer_code=ChatTTS.Chat.InferCodeParams(
spk_emb=spk,
),
)
logger.info("Inference completed.")
# Save each generated wav file to a local file
for index, wav in enumerate(wavs):
@@ -55,9 +58,15 @@ def main(texts: List[str], spk: Optional[str] = None):
if __name__ == "__main__":
logger.info("Starting ChatTTS commandline demo...")
parser = argparse.ArgumentParser(
description="ChatTTS Command", usage='[--spk xxx] "Your text 1." " Your text 2."'
description="ChatTTS Command",
usage='[--spk xxx] "Your text 1." " Your text 2."',
)
parser.add_argument(
"--spk",
help="Speaker (empty to sample a random one)",
type=Optional[str],
default=None,
)
parser.add_argument("--spk", help="Speaker (empty to sample a random one)", type=Optional[str], default=None)
parser.add_argument(
"texts", help="Original text", default="YOUR TEXT HERE", nargs="*"
)