Update app.py

app.py

@@ -18,7 +18,7 @@ def generate_music_tensors(descriptions, duration_seconds: int):
 
     model.set_generation_params(
         use_sampling=True,
-        top_k=250,
+        top_k=256,
         duration=duration_seconds
     )
 
@@ -71,9 +71,11 @@ def main():
         selected_genre = st.selectbox("Select Genre", genres)
         
         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)
 
-        st.subheader("2. Select time duration (In Seconds)")
-        time_slider = st.slider("Select time duration (In Seconds)", 0, 60, 10)
+        # Convert the selected minutes to seconds
+        duration_seconds = time_slider * 60
 
     st.title("""🎵 Song Lab AI 🎵""")
     st.text('')