Create app.py

app.py

@@ -0,0 +1,30 @@
+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)