import torch
from TTS.api import TTS
import gradio as gr


device = "cuda" if torch.cuda.is_available() else "cpu"

def generate_audio(text="And those who were seen dancing were tought to be insane by those who could not hear the music"):

    tts = TTS(model_name='tts_models/en/ljspeech/fast_pitch').to(device)
    tts.tts_to_file(text=text, file_path="outputs/output.wav")

    return "outputs/output.wav"

demo = gr.Interface(fn=generate_audio, 
                    inputs=[gr.Text(label="Text")], 
                    outputs=[gr.Audio(label="Audio")])


demo.launch()