Update app.py

app.py

@@ -13,13 +13,13 @@ 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
+        duration=duration_seconds
     )
 
     with st.spinner("Generating Music..."):