Update app.py

app.py

@@ -17,11 +17,12 @@ def generate_music_tensors(descriptions, duration: int):
     model = load_model()
 
     model.set_generation_params(
-        use_sampling=True,
-        top_k=250,
-        duration=duration
+    use_sampling=True,
+    top_k=250,
+    duration=duration_seconds
     )
 
+
     with st.spinner("Generating Music..."):
         output = model.generate(
             descriptions=descriptions,
@@ -71,7 +72,13 @@ def main():
         selected_genre = st.selectbox("Select Genre", genres)
         
         st.subheader("2. Select time duration (In Seconds)")
-        time_slider = st.slider("Select time duration (In Seconds)", 0, 300, 10)
+        # 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
+        duration_seconds = time_slider * 60
+
+        # time_slider = st.slider("Select time duration (In Seconds)", 0, 300, 10)
 
     st.title("""🎵 Song Lab AI 🎵""")
     st.text('')