Update app.py

app.py

@@ -1,12 +1,10 @@
 import gradio as gr
 import torchaudio
 from audiocraft.models import MusicGen
-from audiocraft.data.audio import audio_write
 import spaces
 import logging
-import os 
-import uuid
 import tempfile
+import numpy as np
 
 # Configura o logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -38,15 +36,13 @@ def generate_music(description, melody_audio):
     logging.info(f"A forma do tensor de áudio gerado: {wav[0].shape}")
     logging.info("Música gerada com sucesso.")
 
-    # Salva o áudio gerado em um arquivo temporário
-    with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as temp_file:
-        torchaudio.save(temp_file.name, wav[0], model.sample_rate)
-        return temp_file.name
+    # Retorna o áudio gerado como uma tupla (sample rate, numpy array)
+    return model.sample_rate, wav[0].cpu().numpy()
 
 # Define a interface Gradio
 description = gr.Textbox(label="Description", placeholder="acoustic, guitar, melody, trap, d minor, 90 bpm")
 melody_audio = gr.Audio(label="Melody Audio (optional)", type="filepath")
-output_audio = gr.Audio(label="Generated Music", type="filepath")
+output_audio = gr.Audio(label="Generated Music", type="numpy")
 
 gr.Interface(
     fn=generate_music,