artificialguybr commited on
Commit
a4709e0
1 Parent(s): c9f6306

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ def generate_music(description, melody_audio):
37
  filename = f'{str(uuid.uuid4())}.wav'
38
  output_path = os.path.join('./', filename) # Salva o arquivo no diretório atual
39
  logging.info(f"Salvando a música gerada em: {output_path}")
40
- audio_write(output_path, wav[0].cpu(), model.sample_rate, strategy="loudness", loudness_compressor=True)
41
 
42
  # Verifica a forma do tensor de áudio e se foi salvo corretamente
43
  logging.info(f"A forma do tensor de áudio gerado: {wav[0].shape}")
 
37
  filename = f'{str(uuid.uuid4())}.wav'
38
  output_path = os.path.join('./', filename) # Salva o arquivo no diretório atual
39
  logging.info(f"Salvando a música gerada em: {output_path}")
40
+ audio_write(output_path, wav[0].cpu().to(torch.float32), model.sample_rate, strategy="loudness", loudness_compressor=True)
41
 
42
  # Verifica a forma do tensor de áudio e se foi salvo corretamente
43
  logging.info(f"A forma do tensor de áudio gerado: {wav[0].shape}")