Update app.py

app.py

@@ -6,6 +6,7 @@ import spaces
 import logging
 import os 
 import uuid
+
 # Configura o logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
@@ -31,28 +32,22 @@ def generate_music(description, melody_audio):
     else:
         logging.info("Gerando música de forma incondicional.")
         wav = model.generate_unconditional(1)
-    filename = f'{str(uuid.uuid4())}.wav'
-    output_path = os.path.join('./', filename)  # Salva o arquivo no diretório atual
-    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 a forma do tensor de áudio e se foi salvo corretamente
+
+    # Verifica a forma do tensor de áudio gerado
     logging.info(f"A forma do tensor de áudio gerado: {wav[0].shape}")
-    logging.info("Música gerada e salva com sucesso.")
-    if not os.path.exists(output_path):
-        raise ValueError(f'Failed to save audio to {output_path}')
+    logging.info("Música gerada com sucesso.")
+
+    return wav[0].cpu().numpy()
 
-    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", value=None)
 
 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=[