fix get_duration

app.py

@@ -121,7 +121,15 @@ def send_msgs(msgs):
 def get_duration(model_name, tab, mid_seq, continuation_state, continuation_select, instruments, drum_kit, bpm,
                  time_sig, key_sig, mid, midi_events, reduce_cc_st, remap_track_channel, add_default_instr,
                  remove_empty_channels, seed, seed_rand, gen_events, temp, top_p, top_k, allow_cc):
-    t = 1e-4*gen_events**2 + 25
+    if tab == 0:
+        start_events = 1 + len(instruments)
+    elif tab == 1 and mid is not None:
+        start_events = midi_events
+    elif tab == 2 and mid_seq is not None:
+        start_events = len(mid_seq[0])
+    else:
+        start_events = 1
+    t = 8.5e-5 * (gen_events+start_events) ** 2 - 8.5e-5 * start_events ** 2 + 23
     if "large" in model_name:
         t *= 2
     return t