Updating requirements and app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 from transformers import pipeline
 import scipy.io.wavfile
 import tempfile
+import numpy as np
 
 
 # -----------------------------------------------------------
@@ -29,11 +30,19 @@ def main():
         if user_text:
             music = synthesizer(user_text, forward_params={"do_sample": True})
 
-            with tempfile.NamedTemporaryFile(delete=True, suffix=".wav") as tmp:
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp:
                 print(f"Temporary file created at: {tmp.name}")
-                scipy.io.wavfile.write(
-                    tmp.name, rate=music["sampling_rate"], data=music["audio"]
-                )
+                # Ensure the audio data is in the correct format (int16 for WAV)
+                if music["audio"].dtype != np.int16:
+                    # Assuming the audio is normalized between -1 and 1, scale to int16
+                    audio_data = np.int16(music["audio"] / np.max(np.abs(music["audio"])) * 32767)
+                else:
+                    audio_data = music["audio"]
+                
+                # Write the audio data to the temporary file
+                scipy.io.wavfile.write(tmp.name, rate=music["sampling_rate"], data=audio_data)
+                
+                # Display the audio player in Streamlit
                 st.audio(tmp.name, format="audio/wav", start_time=0)
 
     except Exception as e:

requirements.txt

@@ -1,3 +1,4 @@
 transformers
 scipy
 torch
+numpy