add optimise midi

app.py

@@ -121,7 +121,7 @@ def send_msgs(msgs):
     return json.dumps(msgs)
 
 
-def run(model_name, tab, instruments, drum_kit, bpm, mid, midi_events, seed, seed_rand,
+def run(model_name, tab, instruments, drum_kit, bpm, mid, midi_events, midi_opt, seed, seed_rand,
         gen_events, temp, top_p, top_k, allow_cc):
     mid_seq = []
     bpm = int(bpm)
@@ -153,7 +153,8 @@ def run(model_name, tab, instruments, drum_kit, bpm, mid, midi_events, seed, see
             disable_patch_change = True
             disable_channels = [i for i in range(16) if i not in patches]
     elif mid is not None:
-        mid = tokenizer.tokenize(MIDI.midi2score(mid))
+        eps = 4 if midi_opt else 0
+        mid = tokenizer.tokenize(MIDI.midi2score(mid), cc_eps=eps, tempo_eps=eps)
         mid = np.asarray(mid, dtype=np.int64)
         mid = mid[:int(midi_events)]
         for token_seq in mid:
@@ -305,6 +306,7 @@ if __name__ == "__main__":
                 input_midi_events = gr.Slider(label="use first n midi events as prompt", minimum=1, maximum=512,
                                               step=1,
                                               value=128)
+                input_midi_opt = gr.Checkbox(label="optimise midi", value=True)
                 example2 = gr.Examples([[file, 128] for file in glob.glob("example/*.mid")],
                                        [input_midi, input_midi_events])
 
@@ -328,8 +330,8 @@ if __name__ == "__main__":
         output_audio = gr.Audio(label="output audio", format="mp3", elem_id="midi_audio")
         output_midi = gr.File(label="output midi", file_types=[".mid"])
         run_event = run_btn.click(run, [input_model, tab_select, input_instruments, input_drum_kit, input_bpm,
-                                        input_midi, input_midi_events, input_seed, input_seed_rand, input_gen_events,
-                                        input_temp, input_top_p, input_top_k, input_allow_cc],
+                                        input_midi, input_midi_events, input_midi_opt, input_seed, input_seed_rand,
+                                        input_gen_events, input_temp, input_top_p, input_top_k, input_allow_cc],
                                   [output_midi_seq, output_midi, output_audio, input_seed, js_msg],
                                   concurrency_limit=3)
         stop_btn.click(cancel_run, [output_midi_seq], [output_midi, output_audio, js_msg], cancels=run_event, queue=False)

requirements.txt

@@ -1,5 +1,5 @@
 Pillow
 numpy
 onnxruntime-gpu
-gradio==4.36.1
+gradio==4.43.0
 pyfluidsynth