Spaces:
Sleeping
Sleeping
| import os | |
| import torch | |
| from InferenceInterfaces.ToucanTTSInterface import ToucanTTSInterface | |
| def read_texts(model_id, sentence, filename, device="cpu", language="en", speaker_reference=None, faster_vocoder=False): | |
| tts = ToucanTTSInterface(device=device, tts_model_path=model_id, faster_vocoder=faster_vocoder) | |
| tts.set_language(language) | |
| if speaker_reference is not None: | |
| tts.set_utterance_embedding(speaker_reference) | |
| if type(sentence) == str: | |
| sentence = [sentence] | |
| tts.read_to_file(text_list=sentence, file_location=filename) | |
| del tts | |
| def the_raven(version, model_id="Meta", exec_device="cpu", speed_over_quality=True, speaker_reference=None): | |
| os.makedirs("audios", exist_ok=True) | |
| read_texts(model_id=model_id, | |
| sentence=['Súru laustanéro taurelasselindon; ondolin ninqanéron, Silmeráno tindon.'], | |
| filename=f"audios/{version}.wav", | |
| device=exec_device, | |
| language="qya", | |
| speaker_reference=speaker_reference, | |
| faster_vocoder=speed_over_quality) | |
| if __name__ == '__main__': | |
| exec_device = "cuda" if torch.cuda.is_available() else "cpu" | |
| print(f"running on {exec_device}") | |
| the_raven(version="fi-oilimamarkiryai-03.wav", | |
| model_id="/Users/zhou/Desktop/Quenya-TTS/Models/ToucanTTS_Quenya/6k_Finnish.pt", # Finnish English | |
| exec_device=exec_device, | |
| speed_over_quality=exec_device != "cuda") | |