Update app.py

app.py

@@ -1,12 +1,10 @@
 import streamlit as st
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-import torchaudio
+from transformers import AutoProcessor, AutoModelForCausalLM
 from audiocraft.models import MusicGen
-from audiocraft.data.audio import audio_write
 
 @st.cache_resource
 def load_musicgen_model():
-    model = MusicGen.get_pretrained('melody')  
+    model = MusicGen.get_pretrained("facebook/musicgen-large")  
     model.set_generation_params(duration=10)  
     return model
 
@@ -27,8 +25,6 @@ if st.button("Generate Song"):
     
     audio_output = model.generate([description])
     
-    audio_write("generated_song.wav", audio_output[0].cpu(), model.sample_rate)
-    
-    st.audio("generated_song.wav", format="audio/wav")
+    st.audio(audio_output[0].cpu().detach().numpy(), format="audio/wav")
 
 st.write("Enjoy your personalized music experience! 🎼")