ford442 commited on
Commit
59743f4
·
verified ·
1 Parent(s): 7f6755b

Update demos/musicgen_app.py

Browse files
Files changed (1) hide show
  1. 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], 32000, strategy="loudness", #hardcoded sample rate
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], 32000, strategy="loudness", #hardcoded sample rate
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)