tfg-demo / app.py
daniel-dona's picture
Solve memory problem
3d98234
raw
history blame
777 Bytes
import gradio as gr
import os
from TTS.config import load_config
from TTS.utils.manage import ModelManager
from TTS.utils.synthesizer import Synthesizer
model_path = "./model/model.pth"
config_path = "./model/config.json"
output_path = "/tmp/"
def run_tts(text):
synthesizer = Synthesizer(
tts_checkpoint=model_path,
tts_config_path=config_path,
tts_speakers_file=None,
tts_languages_file=None,
vocoder_checkpoint=None,
vocoder_config=None,
encoder_checkpoint="",
encoder_config="",
use_cuda=False,
)
wav = synthesizer.tts(text)
output_file = os.path.join(output_path, "inference.wav")
synthesizer.save_wav(wav, output_file)
return output_file
iface = gr.Interface(fn=run_tts, inputs="text", outputs="audio", enable_queue=True)
iface.launch()