Update app.py

app.py

@@ -1,17 +1,13 @@
 import gradio as gr
-import torchaudio  
+import torchaudio
 from audiocraft.models import MusicGen
-from audiocraft.data.audio import audio_write
 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')
 
-@spaces.GPU(duration=120)  
+@spaces.GPU(duration=120)
 def generate_music(description, melody_audio):
     logging.info("Iniciando a geração de música.")
 
@@ -33,39 +29,25 @@ def generate_music(description, melody_audio):
     else:
         logging.info("Gerando música de forma incondicional.")
         wav = model.generate_unconditional(1)
-        
-    # Gera nome de arquivo único 
-    filename = f'{str(uuid.uuid4())}.wav'
-    
-    # Salva na pasta temporária
-    output_path = tempfile.mktemp(suffix='.wav')  
-    logging.info(f"Salvando a música gerada em: {output_path}")
-    
-    audio_write(output_path, wav[0].cpu(), model.sample_rate, 
-                strategy="loudness", loudness_compressor=True)
-    
-    # Verifica se o arquivo existe
-    if not os.access(output_path, os.R_OK):
-        raise ValueError(f'Failed to save audio to {output_path}')
 
     logging.info(f"A forma do tensor de áudio gerado: {wav[0].shape}")
-    logging.info("Música gerada e salva com sucesso.")
+    logging.info("Música gerada com sucesso.")
+
+    return wav[0]  # Retorna o tensor de áudio diretamente
 
-    return output_path
-    
 # 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_path = gr.Audio(label="Generated Music", type="filepath")
+output_audio = gr.Audio(label="Generated Music", type="numpy")  # Especifica o tipo como "numpy"
 
 gr.Interface(
     fn=generate_music,
     inputs=[description, melody_audio],
-    outputs=output_path,
+    outputs=output_audio,
     title="MusicGen Demo",
     description="Generate music using the MusicGen model.",
     examples=[
         ["trap, synthesizer, songstarters, dark, G# minor, 140 bpm", "./assets/kalhonaho.mp3"],
         ["upbeat, electronic, synth, dance, 120 bpm", None]
     ]
-).launch()
+).launch()