Added main function as entry point

This commit is contained in:
Heiko J Schick
2022-08-30 16:28:44 +02:00
parent 5978cb7bcf
commit 494b15fd91
+22 -18
View File
@@ -10,25 +10,29 @@ import scipy
import numpy as np
import IPython.display as ipd
def main():
"""
Defined starting point of source code.
"""
models, cfg, task = load_model_ensemble_and_task_from_hf_hub(
models, cfg, task = load_model_ensemble_and_task_from_hf_hub(
"facebook/fastspeech2-en-ljspeech",
arg_overrides={"vocoder": "hifigan", "fp16": False}
)
)
TTSHubInterface.update_cfg_with_data_cfg(cfg, task.data_cfg)
generator = task.build_generator(models, cfg)
TTSHubInterface.update_cfg_with_data_cfg(cfg, task.data_cfg)
generator = task.build_generator(models, cfg)
full_wave_file = []
rate = 44100
sentences = []
full_wave_file = []
rate = 44100
sentences = []
# Read input file
with open(f"input.txt", "r") as f:
# Read input file
with open(f"input.txt", "r") as f:
lines = f.readlines()
# Convert to sentences
for line in lines:
# Convert to sentences
for line in lines:
line = line.replace("-", " - ")
line = line.replace("/", ", ")
line = line.replace("", ". ")
@@ -37,14 +41,11 @@ for line in lines:
line = line.replace("(", ". ")
line = line.replace(")", ". ")
for x in line.split(". "):
# print(x)
sentences.append(x.strip())
# print(sentences)
sentences.append("<PAUSE>")
# print(sentences)
# Synthesis text
for text in sentences:
# Synthesis text
for text in sentences:
if text == "":
continue
@@ -58,5 +59,8 @@ for text in sentences:
wav = wav.numpy()
full_wave_file.extend(wav)
full_wave_file = np.array(full_wave_file, dtype=np.float32)
scipy.io.wavfile.write("test.wav", rate, full_wave_file)
full_wave_file = np.array(full_wave_file, dtype=np.float32)
scipy.io.wavfile.write("test.wav", rate, full_wave_file)
if __name__ == "__main__":
main()