Update app.py

app.py

@@ -176,10 +176,14 @@ def run(model_name, tab, instruments, drum_kit, mid, midi_events, gen_events, te
     generator = generate(model, mid, max_len=max_len, temp=temp, top_p=top_p, top_k=top_k,
                          disable_patch_change=disable_patch_change, disable_control_change=not allow_cc,
                          disable_channels=disable_channels)
-    for token_seq in generator:
-        mid_seq.append(token_seq)
-        draw_event(token_seq)
-        yield mid_seq, get_img(), None, None
+    try:
+        for token_seq in generator:
+            mid_seq.append(token_seq)
+            draw_event(token_seq)
+            yield mid_seq, get_img(), None, None
+    except Exception as e:
+        print(e)
+
     mid = tokenizer.detokenize(mid_seq)
     with open(f"output.mid", 'wb') as f:
         f.write(MIDI.score2midi(mid))