Update app.py

app.py

@@ -6,6 +6,7 @@ import spaces
 import logging
 import os 
 import uuid
+import tempfile
 
 # Configura o logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -37,12 +38,15 @@ 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.")
 
-    return wav[0].cpu().numpy()
+    # 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
 
 # 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", value=None)
+output_audio = gr.Audio(label="Generated Music", type="filepath")
 
 gr.Interface(
     fn=generate_music,