unsloth-tts-api / app.py
Alex1957's picture
Create app.py
4e1b0db verified
Raw
History Blame Contribute Delete
490 Bytes
import gradio as gr
from transformers import pipeline
# Load your fine-tuned model from HF Hub or local directory
tts_pipeline = pipeline("text-to-speech", model="Unsloth/sesame-csm-1b-tts")
def generate_speech(text):
output = tts_pipeline(text)
return output["audio"]
iface = gr.Interface(
fn=generate_speech,
inputs=gr.Textbox(lines=2, placeholder="Enter text..."),
outputs=gr.Audio(type="numpy", label="Generated Speech"),
title="Sesame TTS"
)
iface.launch()