File size: 1,120 Bytes
c9abc7d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
import gradio as gr
from transformers import AutoProcessor, AudioSpectrogramPipeline
import soundfile as sf
import tempfile
# โหลดโมเดลและ processor
processor = AutoProcessor.from_pretrained("stabilityai/stable-audio-open-small")
pipeline = AudioSpectrogramPipeline.from_pretrained("stabilityai/stable-audio-open-small")
def generate_audio(prompt):
result = pipeline(prompt=prompt, num_inference_steps=50)
audio = result["audio"]
sr = result["sampling_rate"]
# บันทึกไฟล์ชั่วคราวเพื่อให้ Gradio เล่นได้
with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as f:
sf.write(f.name, audio, sr)
return f.name
# สร้าง UI
demo = gr.Interface(
fn=generate_audio,
inputs=gr.Textbox(label="Prompt (in English)", placeholder="e.g. A lo-fi beat with rain sounds"),
outputs=gr.Audio(label="Generated Audio"),
title="🎵 Stable Audio - Text to Sound",
description="Generate short sound clips from text using `stabilityai/stable-audio-open-small`"
)
demo.launch()
|