fix filename issue

#1
by multimodalart HF staff - opened
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -35,17 +35,17 @@ def generate_music(description, melody_audio):
35
  else:
36
  logging.info("Gerando música de forma incondicional.")
37
  wav = model.generate_unconditional(1)
38
- filename = f'{str(uuid.uuid4())}.wav'
39
  logging.info(f"Salvando a música gerada com o nome: {filename}")
40
- audio_write(filename, 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}")
44
  logging.info("Música gerada e salva com sucesso.")
45
- if not os.path.exists(filename):
46
- raise ValueError(f'Failed to save audio to {filename}')
47
 
48
- return filename
49
 
50
  # Define a interface Gradio
51
  description = gr.Textbox(label="Description", placeholder="acoustic, guitar, melody, trap, d minor, 90 bpm")
 
35
  else:
36
  logging.info("Gerando música de forma incondicional.")
37
  wav = model.generate_unconditional(1)
38
+ filename = f'{str(uuid.uuid4())}'
39
  logging.info(f"Salvando a música gerada com o nome: {filename}")
40
+ path = audio_write(filename, wav[0].cpu().to(torch.float32), model.sample_rate, strategy="loudness", loudness_compressor=True)
41
+ print("Música salva em", path, ".")
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}")
44
  logging.info("Música gerada e salva com sucesso.")
45
+ if not os.path.exists(path):
46
+ raise ValueError(f'Failed to save audio to {path}')
47
 
48
+ return path
49
 
50
  # Define a interface Gradio
51
  description = gr.Textbox(label="Description", placeholder="acoustic, guitar, melody, trap, d minor, 90 bpm")