fvbarros commited on
Commit
53cc30b
·
verified ·
1 Parent(s): 9d010f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -14,16 +14,16 @@ audiogen.set_generation_params(duration=2)
14
 
15
  def generate_music(prompt):
16
  print("Gerando música para:", prompt)
17
- wav = musicgen.generate(prompt)
18
  print("Áudio gerado, salvando...")
19
  with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as fp:
20
- audio_write(fp.name, wav.cpu(), musicgen.sample_rate, strategy="loudness")
21
  print("Arquivo salvo em", fp.name)
22
  return fp.name
23
 
24
  def generate_sound_effect(prompt):
25
  print("Gerando som para:", prompt)
26
- wav = audiogen.generate(prompt)
27
  print("Som gerado, salvando...")
28
  with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as fp:
29
  audio_write(fp.name, wav.cpu(), audiogen.sample_rate, strategy="loudness", loudness_compressor=True)
 
14
 
15
  def generate_music(prompt):
16
  print("Gerando música para:", prompt)
17
+ wav = musicgen.generate([prompt])[0]
18
  print("Áudio gerado, salvando...")
19
  with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as fp:
20
+ audio_write(fp.name, wav.cpu(), musicgen.sample_rate, strategy="loudness", loudness_compressor=True)
21
  print("Arquivo salvo em", fp.name)
22
  return fp.name
23
 
24
  def generate_sound_effect(prompt):
25
  print("Gerando som para:", prompt)
26
+ wav = audiogen.generate([prompt])[0]
27
  print("Som gerado, salvando...")
28
  with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as fp:
29
  audio_write(fp.name, wav.cpu(), audiogen.sample_rate, strategy="loudness", loudness_compressor=True)