RohitCSharp commited on
Commit
fd34583
Β·
verified Β·
1 Parent(s): ae44623

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -21,13 +21,14 @@ def generate_music(prompt):
21
  sampling_rate = model.config.audio_encoder.sampling_rate
22
  audio = audio_values[0].cpu().numpy()
23
 
24
- # Normalize and convert to int16
25
- audio = audio / max(abs(audio)) # Normalize to -1.0 to 1.0
26
- audio = (audio * 32767).astype(np.int16)
27
 
28
- # Save to WAV
29
  output_path = "output.wav"
30
- scipy.io.wavfile.write(output_path, sampling_rate, audio)
 
31
 
32
  return (sampling_rate, audio)
33
 
 
21
  sampling_rate = model.config.audio_encoder.sampling_rate
22
  audio = audio_values[0].cpu().numpy()
23
 
24
+ # Normalize to float32 in -1.0 to 1.0 range (Gradio requirement for type="numpy")
25
+ audio = audio / np.max(np.abs(audio))
26
+ audio = audio.astype(np.float32)
27
 
28
+ # Also save as .wav (optional)
29
  output_path = "output.wav"
30
+ int_audio = (audio * 32767).astype(np.int16)
31
+ scipy.io.wavfile.write(output_path, sampling_rate, int_audio)
32
 
33
  return (sampling_rate, audio)
34