Update app.py

app.py

@@ -17,7 +17,7 @@ genres = ["Pop", "Rock", "Jazz", "Electronic", "Hip-Hop", "Classical", "Lofi", "
 
 @st.cache_resource()
 def load_model():
-    model = MusicGen.get_pretrained('facebook/musicgen-large')
+    model = MusicGen.get_pretrained('facebook/musicgen-small')
     return model
 
 def generate_music_tensors(descriptions, duration: int):
@@ -68,7 +68,7 @@ st.set_page_config(
 )
 
 def main():
-    st.title("🎧 AI Composer Medium-Model 🎧")
+    st.title("🎧 AI Composer Small-Model 🎧")
 
     st.subheader("Craft your perfect melody!")
     bpm = st.number_input("Enter Speed in BPM", min_value=60)
@@ -88,14 +88,15 @@ def main():
     if st.button('Let\'s Generate 🎶'):
         st.text('\n\n')
         st.subheader("Generated Music")
+        descriptions = [f"{text_area} {selected_genre} {bpm} BPM" for _ in range(1)]  # Change the batch size to 1
         
         # Generate audio
-        description = text_area  # Initialize description with text_area
-        if selected_genre:
-            description += f" {selected_genre}"
-            st.empty()  # Hide the selected_genre selectbox after selecting one option
-        if bpm:
-            description += f" {bpm} BPM"
+        # description = text_area  # Initialize description with text_area
+        # if selected_genre:
+        #     description += f" {selected_genre}"
+        #     st.empty()  # Hide the selected_genre selectbox after selecting one option
+        # if bpm:
+        #     description += f" {bpm} BPM"
         # if mood:
         #     description += f" {mood}"
         #     st.empty()  # Hide the mood selectbox after selecting one option