Update app.py

app.py

@@ -49,9 +49,11 @@ def synthesize(midi_file=None):
 
 def run_example(midi_file=None, prompt="", neg_prompt="", duration=None, seed=0, cond=1, inf=20, guidance_scale=2.5, guess=False):
     midi_synth = synthesize(midi_file)
-    if duration == 0:
+    if not duration or duration == 0:
+        print("Duration", duration)
         duration = midi_synth[1].shape[0]/SAMPLE_RATE
-    gen_audio = predict(midi_file, prompt, neg_prompt, duration, seed, cond, inf, guidance_scale, guess)
+        print("New duration", duration)
+    gen_audio = predict(midi_file, prompt, neg_prompt, duration=duration, seed, cond, inf, guidance_scale, guess)
     return midi_synth, gen_audio
 
 with gr.Blocks(title="🎹 MIDI-AudioLDM", theme=gr.themes.Base(text_size=gr.themes.sizes.text_md, font=[gr.themes.GoogleFont("Nunito Sans")])) as demo: