Added main function as entry point
This commit is contained in:
@@ -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()
|
||||
Reference in New Issue
Block a user