| import gradio as gr | |
| import soundfile as sf | |
| from auffusion_pipeline import AuffusionPipeline | |
| pipeline = AuffusionPipeline.from_pretrained("auffusion/auffusion") | |
| def infer(prompt): | |
| prompt = "Birds singing sweetly in a blooming garden" | |
| output = pipeline(prompt=prompt) | |
| audio = output.audios[0] | |
| sf.write(f"{prompt}.wav", audio, samplerate=16000) | |
| return f"{prompt}.wav" | |
| with gr.Blocks() as demo: | |
| with gr.Column(): | |
| gr.Markdown("# Auffusion") | |
| prompt = gr.Textbox(label="Prompt") | |
| submit_btn = gr.Button("Submit") | |
| audio_out = gr.Audio(label="Audio Ressult") | |
| submit_btn.click( | |
| fn = infer, | |
| inputs = [prompt], | |
| outputs = [audio_out] | |
| ) | |
| demo.queue().launch(show_api=False, show_error=True) |