Update app.py

app.py

@@ -1,19 +1,32 @@
 import gradio as gr
-from audiocraft.models import AudioGen
+from audiocraft.models import MusicGen, 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)
+# Carregar modelos
+musicgen = MusicGen.get_pretrained('facebook/musicgen-small')
+audiogen = AudioGen.get_pretrained('facebook/audiogen-medium')
 
-def generate_audio(prompt):
-    wav = model.generate([prompt])[0]
+# Definir duração padrão
+musicgen.set_generation_params(duration=5)
+audiogen.set_generation_params(duration=5)
+
+def generate_music(prompt):
+    wav = musicgen.generate([prompt])[0]
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as fp:
+        audio_write(fp.name, wav.cpu(), musicgen.sample_rate)
+        return fp.name
+
+def generate_sound_effect(prompt):
+    wav = audiogen.generate([prompt])[0]
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as fp:
-        audio_write(fp.name, wav.cpu(), model.sample_rate)
+        audio_write(fp.name, wav.cpu(), audiogen.sample_rate)
         return fp.name
 
-demo = gr.Interface(fn=generate_audio, inputs=gr.Text(label="Prompt de áudio"), outputs=gr.Audio(type="filepath"))
+music_interface = gr.Interface(fn=generate_music, inputs=gr.Text(label="Prompt para música"), outputs=gr.Audio(type="filepath"), title="🎵 MusicGen")
+sound_interface = gr.Interface(fn=generate_sound_effect, inputs=gr.Text(label="Prompt para efeito sonoro"), outputs=gr.Audio(type="filepath"), title="🔊 AudioGen")
 
+# Interface combinada
+demo = gr.TabbedInterface([music_interface, sound_interface], ["MusicGen", "AudioGen"])
 demo.launch()