Update app.py

app.py

@@ -47,8 +47,10 @@ def synthesize(midi_file=None):
     midi_synth = midi_synth.reshape(midi_synth.shape[0], 1)
     return (SAMPLE_RATE, midi_synth)
 
-def run_example(midi_file=None, prompt="", neg_prompt="", duration=10, seed=0, cond=1, inf=20, guidance_scale=2.5, guess=False):
+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 not duration:
+        duration = midi_synth[1].shape[0]/SAMPLE_RATE
     gen_audio = predict(midi_file, prompt, neg_prompt, duration, seed, cond, inf, guidance_scale, guess)
     return midi_synth, gen_audio