Update app.py

app.py

@@ -8,6 +8,7 @@ import os
 import uuid
 from torch.cuda.amp import autocast
 import torch
+
 ZERO_GPU_PATCH_TORCH_DEVICE = 1
 
 # Configura o logging
@@ -17,6 +18,10 @@ logging.info("Carregando o modelo pré-treinado.")
 model = MusicGen.get_pretrained('nateraw/musicgen-songstarter-v0.2')
 model.set_generation_params(duration=8)
 
+# Define o diretório de saída para salvar os arquivos de áudio gerados
+output_dir = './generated_music'
+os.makedirs(output_dir, exist_ok=True)
+
 @spaces.GPU(duration=120)
 def generate_music(description, melody_audio):
     with autocast():
@@ -35,7 +40,7 @@ def generate_music(description, melody_audio):
             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
+        output_path = os.path.join(output_dir, filename)  # Salva o arquivo no diretório de saída
         logging.info(f"Salvando a música gerada em: {output_path}")
         audio_write(output_path, wav[0].cpu().to(torch.float32), model.sample_rate, strategy="loudness", loudness_compressor=True)