import gradio as gr
from transformers import pipeline

# Load the TTS model
model = pipeline("text-to-speech", model="galsenai/parler-tts-mini-v1-wolof")

def text_to_speech(text):
    audio = model(text)
    return audio

# Create the Gradio interface
interface = gr.Interface(fn=text_to_speech, inputs="text", outputs="audio")

interface.launch()