Spaces:
Build error
Build error
| import gradio as gr | |
| import time | |
| import urllib.request | |
| from pathlib import Path | |
| import os | |
| import torch | |
| import scipy.io.wavfile | |
| from espnet2.bin.tts_inference import Text2Speech | |
| from espnet2.utils.types import str_or_none | |
| gos_text2speech = Text2Speech.from_pretrained( | |
| model_tag="https://huggingface.co/wietsedv/tacotron2-gronings/resolve/main/tts_ljspeech_finetune_tacotron2.v5_train.loss.ave.zip", | |
| vocoder_tag="parallel_wavegan/ljspeech_parallel_wavegan.v3", | |
| device="cpu", | |
| threshold=0.5, | |
| minlenratio=0.0, | |
| maxlenratio=10.0, | |
| use_att_constraint=True, | |
| backward_window=1, | |
| forward_window=4, | |
| ) | |
| def inference(text,lang): | |
| with torch.no_grad(): | |
| if lang == "gronings": | |
| wav = gos_text2speech(text)["wav"] | |
| scipy.io.wavfile.write("out.wav", gos_text2speech.fs , wav.view(-1).cpu().numpy()) | |
| return "out.wav", "out.wav" | |
| title = "GroTTS" | |
| examples = [ | |
| ['Ze gingen mit klas noar waddendiek, over en deur bragel lopen.', 'gronings'] | |
| ] | |
| gr.Interface( | |
| inference, | |
| [gr.inputs.Textbox(label="input text", lines=3), gr.inputs.Radio(choices=["gronings"], type="value", default="gronings", label="language")], | |
| [gr.outputs.Audio(type="file", label="Output"), gr.outputs.File()], | |
| title=title, | |
| examples=examples | |
| ).launch(enable_queue=True) | |