storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-17 00:23:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update fakeyou_infer.py

Browse Source
This commit is contained in:
Justin John
2024-01-11 13:53:33 +05:30
committed by GitHub
parent 426aba80d6
commit 7c22f04664
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tts/StyleTTS2/fakeyou_infer.py
+8 -4
View File
@@ -6,7 +6,11 @@ import soundfile as sf
from styletts2importable import compute_style, inference
from tortoise.utils.text import split_and_recombine_text
def clsynthesize(text, voice_path, vcsteps):
def clsynthesize(text_file_path, voice_path, vcsteps):
# Load the text from the file
with open(text_file_path, 'r', encoding="utf-8") as file:
text = file.read()
# Load the custom voice
custom_voice = compute_style(voice_path)
@@ -22,16 +26,16 @@ def clsynthesize(text, voice_path, vcsteps):
def main():
parser = argparse.ArgumentParser(description='CLI for StyleTTS Voice Synthesis')
parser.add_argument('--text', type=str, required=True, help='Text to synthesize')
parser.add_argument('--input', type=str, required=True, help='Path to the text file')
parser.add_argument('--voice', type=str, required=True, help='Path to the custom voice audio file')
parser.add_argument('--vcsteps', type=int, required=True, help='Number of diffusion steps')
parser.add_argument('--output', type=str, required=True, help='Output path for the synthesized audio file')
args = parser.parse_args()
sample_rate, audio = clsynthesize(args.text, args.voice, args.vcsteps)
sample_rate, audio = clsynthesize(args.input, args.voice, args.vcsteps)
# Save the synthesized audio to the specified output path
sf.write(args.output, audio, sample_rate)
if __name__ == "__main__":
main()
main()