Update app.py

app.py

@@ -18,6 +18,11 @@ def generate_music(prompt):
     audio_values = model.generate(**inputs, max_new_tokens=256)
     sampling_rate = model.config.audio_encoder.sampling_rate
     audio = audio_values[0].cpu().numpy()
+
+    # Normalize and convert to int16 for playback safety
+    audio = audio / max(abs(audio))  # Normalize to -1.0 to 1.0
+    audio = (audio * 32767).astype("int16")  # Convert to int16 format
+
     return (sampling_rate, audio)
 
 # Function to get enhanced prompt via GPT