artificialguybr commited on
Commit
8bdaf42
1 Parent(s): 051ea93

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -17,11 +17,13 @@ def generate_audio(prompt, audio_input=None):
17
  else:
18
  # Carrega o áudio de entrada e gera com base nele e nas descrições
19
  melody, sr = torchaudio.load(audio_input)
 
 
20
  wav = model.generate_with_chroma(descriptions, melody[None], sr)
21
 
22
  # Salva o arquivo de áudio gerado
23
  output_path = 'generated_audio.wav'
24
- audio_write(output_path, wav[0].cpu(), model.sample_rate, strategy="loudness", loudness_compressor=True)
25
 
26
  return output_path
27
 
 
17
  else:
18
  # Carrega o áudio de entrada e gera com base nele e nas descrições
19
  melody, sr = torchaudio.load(audio_input)
20
+ # Assegura que o tensor está no formato de ponto flutuante de precisão simples
21
+ melody = melody.float()
22
  wav = model.generate_with_chroma(descriptions, melody[None], sr)
23
 
24
  # Salva o arquivo de áudio gerado
25
  output_path = 'generated_audio.wav'
26
+ audio_write(output_path, wav[0].cpu().float(), model.sample_rate, strategy="loudness", loudness_compressor=True)
27
 
28
  return output_path
29