update duration params

gradio_components/prediction.py

@@ -40,6 +40,7 @@ def _do_predictions(
     )
     be = time.time()
     processed_melodies = []
+    model.set_generation_params(duration=duration)
     for melody in melodies:
         if melody is None:
             processed_melodies.append(None)
@@ -138,7 +139,7 @@ def predict(
         if INTERRUPTING:
             raise gr.Error("Interrupted.")
 
-    model.set_custom_progress_callback(_progress)
+    model.set_custom_progress_callback(_progress
 
     wavs = _do_predictions(
         model_path,