import gradio as gr
from transformers import pipeline

tts = pipeline("text-to-audio", model="facebook/musicgen-small")

def text_to_audio(text):
    output = tts(text)
    return output["audio"]
iface = gr.Interface(
    fn=text_to_audio,
    inputs=gr.Textbox(label="Text"),
    outputs=gr.Audio(label="Audio"),
    title="Text-to-Audio",

)

if __name__ == "__main__":
    iface.launch()