Update demos/musicgen_app.py
Browse files- demos/musicgen_app.py +2 -2
demos/musicgen_app.py
CHANGED
@@ -256,7 +256,7 @@ def predict_full(model, model_path, use_mbd, text, melody, duration, topk, topp,
|
|
256 |
# Save standard output
|
257 |
with NamedTemporaryFile("wb", suffix=".wav", delete=False) as file:
|
258 |
audio_write(
|
259 |
-
file.name, wav[0],
|
260 |
loudness_headroom_db=16, loudness_compressor=True, add_suffix=False
|
261 |
)
|
262 |
wav_paths.append(file.name)
|
@@ -269,7 +269,7 @@ def predict_full(model, model_path, use_mbd, text, melody, duration, topk, topp,
|
|
269 |
if diffusion_wav is not None:
|
270 |
with NamedTemporaryFile("wb", suffix=".wav", delete=False) as file:
|
271 |
audio_write(
|
272 |
-
file.name, diffusion_wav[0],
|
273 |
loudness_headroom_db=16, loudness_compressor=True, add_suffix=False
|
274 |
)
|
275 |
wav_paths.append(file.name)
|
|
|
256 |
# Save standard output
|
257 |
with NamedTemporaryFile("wb", suffix=".wav", delete=False) as file:
|
258 |
audio_write(
|
259 |
+
file.name, wav[0], 44100, strategy="loudness", #hardcoded sample rate
|
260 |
loudness_headroom_db=16, loudness_compressor=True, add_suffix=False
|
261 |
)
|
262 |
wav_paths.append(file.name)
|
|
|
269 |
if diffusion_wav is not None:
|
270 |
with NamedTemporaryFile("wb", suffix=".wav", delete=False) as file:
|
271 |
audio_write(
|
272 |
+
file.name, diffusion_wav[0], 44100, strategy="loudness", #hardcoded sample rate
|
273 |
loudness_headroom_db=16, loudness_compressor=True, add_suffix=False
|
274 |
)
|
275 |
wav_paths.append(file.name)
|