Update app.py

app.py

@@ -13,16 +13,15 @@ def load_model():
     model = MusicGen.get_pretrained('facebook/musicgen-small')
     return model
 
-def generate_music_tensors(descriptions, duration: int):
+def generate_music_tensors(descriptions, duration_seconds: int):
     model = load_model()
 
     model.set_generation_params(
-    use_sampling=True,
-    top_k=250,
-    duration=duration_seconds
+        use_sampling=True,
+        top_k=250,
+        duration=duration_seconds
     )
 
-
     with st.spinner("Generating Music..."):
         output = model.generate(
             descriptions=descriptions,
@@ -71,15 +70,13 @@ def main():
         # Dropdown for genres
         selected_genre = st.selectbox("Select Genre", genres)
         
-        st.subheader("2. Select time duration (In Seconds)")
+        st.subheader("2. Select time duration (In Minutes)")
         # Adjust the time slider for selecting duration in minutes
         time_slider = st.slider("Select time duration (In Minutes)", 1, 5, 1)
 
-# Convert the selected minutes to seconds
+        # Convert the selected minutes to seconds
         duration_seconds = time_slider * 60
 
-        # time_slider = st.slider("Select time duration (In Seconds)", 0, 300, 10)
-
     st.title("""🎵 Song Lab AI 🎵""")
     st.text('')
     left_co,right_co = st.columns(2)
@@ -98,9 +95,9 @@ def main():
 
             # Generate audio
             descriptions = [f"{text_area} {selected_genre} {bpm} BPM" for _ in range(5)]  # Adjust the batch size (5 in this case)
-            music_tensors = generate_music_tensors(descriptions, time_slider)
+            music_tensors = generate_music_tensors(descriptions, duration_seconds)
 
-             # Only play the full audio for index 0
+            # Only play the full audio for index 0
             idx = 0
             music_tensor = music_tensors[idx]
             save_music_file = save_audio(music_tensor)
@@ -115,4 +112,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-