Update app.py

app.py

@@ -0,0 +1,34 @@
+import streamlit as st
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+import torchaudio
+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.set_generation_params(duration=10)  
+    return model
+
+st.title("Mood-Based Music Generator 🎶")
+st.write("Generate a personalized song based on how you feel today.")
+
+mood = st.selectbox(
+    "Select your mood today:",
+    ("Happy", "Sad", "Energetic", "Calm", "Romantic", "Mysterious")
+)
+
+prompt = st.text_input("What's on your mind?", "Describe a scene or feeling")
+
+model = load_musicgen_model()
+
+if st.button("Generate Song"):
+    description = f"{mood.lower()} {prompt.lower()}"
+    
+    audio_output = model.generate([description])
+    
+    audio_write("generated_song", audio_output[0].cpu(), model.sample_rate)
+    
+    st.audio("generated_song.wav", format="audio/wav")
+
+st.write("Enjoy your personalized music experience! 🎼")