1inkusFace commited on
Commit
044cbb3
·
verified ·
1 Parent(s): c1ef20f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -69,15 +69,15 @@ def generate_audio(prompt, seconds_total=30, steps=100, cfg_scale=7, use_bfloat=
69
  output = rearrange(output, "b d n -> d (b n)")
70
  # Peak normalize, clip, convert to int16
71
  output = output.to(torch.float32).div(torch.max(torch.abs(output))).clamp(-1, 1).mul(32767).to(torch.int16).cpu()
72
- unique_filename = f"output_{uuid.uuid4().hex}.mp3"
73
  print(f"Saving audio to file: {unique_filename}")
74
  torchaudio.save(
75
  unique_filename,
76
  output,
77
  sample_rate,
78
- format="mp3",
79
- encoding="MP3",
80
- bits_per_sample=320
81
  )
82
  print(f"Audio saved: {unique_filename}")
83
  return unique_filename
 
69
  output = rearrange(output, "b d n -> d (b n)")
70
  # Peak normalize, clip, convert to int16
71
  output = output.to(torch.float32).div(torch.max(torch.abs(output))).clamp(-1, 1).mul(32767).to(torch.int16).cpu()
72
+ unique_filename = f"output_{uuid.uuid4().hex}.flac"
73
  print(f"Saving audio to file: {unique_filename}")
74
  torchaudio.save(
75
  unique_filename,
76
  output,
77
  sample_rate,
78
+ format="flac",
79
+ encoding="PCM_F",
80
+ bits_per_sample=32
81
  )
82
  print(f"Audio saved: {unique_filename}")
83
  return unique_filename