duration

app.py

@@ -118,6 +118,9 @@ def send_msgs(msgs):
     return json.dumps(msgs)
 
 
+def calc_time(x):
+    return 5.849e-5*x**2 + 0.04781*x + 0.1168
+
 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):
@@ -129,7 +132,7 @@ def get_duration(model_name, tab, mid_seq, continuation_state, continuation_sele
         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
+    t = calc_time(start_events + gen_events) - calc_time(start_events) + 5
     if "large" in model_name:
         t *= 2
     return t