Spaces:
Running
on
Zero
Running
on
Zero
fix filename issue
#1
by
multimodalart
HF staff
- opened
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())}
|
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(
|
46 |
-
raise ValueError(f'Failed to save audio to {
|
47 |
|
48 |
-
return
|
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")
|