Update app.py

app.py

@@ -0,0 +1,19 @@
+import gradio as gr
+from audiocraft.models import AudioGen
+from audiocraft.data.audio import audio_write
+import torchaudio
+import tempfile
+import os
+
+model = AudioGen.get_pretrained('facebook/audiogen-medium')
+model.set_generation_params(duration=5)
+
+def generate_audio(prompt):
+    wav = model.generate([prompt])[0]
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as fp:
+        audio_write(fp.name, wav.cpu(), model.sample_rate)
+        return fp.name
+
+demo = gr.Interface(fn=generate_audio, inputs=gr.Text(label="Prompt de áudio"), outputs=gr.Audio(type="filepath"))
+
+demo.launch()