| import torch | |
| import soundfile as sf | |
| from .config import pipe_tts | |
| from io import BytesIO | |
| SAMPLING_RATE = 16000 | |
| class T2A: | |
| def __init__(self, input_text: str): | |
| self.output_model = pipe_tts(input_text) | |
| def get_audio(self): | |
| if self.inputs is not None: | |
| synth = self.output_model["audio"][0] | |
| print(f"synth : {synth}") | |
| with BytesIO() as buffer: | |
| sf.write(buffer, synth, SAMPLING_RATE, format='wav') | |
| output = buffer.getvalue() # bytes | |
| print(f"output : {output}, type : {type(output)}") | |
| return output | |
| else: | |
| raise Exception("Input text is None. Please provide text") |