changes

app.py

@@ -1,5 +1,3 @@
-from concurrent.futures import ThreadPoolExecutor
-
 import spaces
 import random
 import argparse
@@ -7,6 +5,7 @@ import glob
 import json
 import os
 import time
+from concurrent.futures import ThreadPoolExecutor
 
 import gradio as gr
 import numpy as np
@@ -122,7 +121,7 @@ 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 + 15
+    t = 1e-4*gen_events**2 + 25
     if "large" in model_name:
         t *= 2
     return t
@@ -383,7 +382,7 @@ if __name__ == "__main__":
     with app:
         gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Midi Composer</h1>")
         gr.Markdown("![Visitors](https://api.visitorbadge.io/api/visitors?path=skytnt.midi-composer&style=flat)\n\n"
-                    "Midi event transformer for music generation\n\n"
+                    "Midi event transformer for symbolic music generation\n\n"
                     "Demo for [SkyTNT/midi-model](https://github.com/SkyTNT/midi-model)\n\n"
                     "[Open In Colab]"
                     "(https://colab.research.google.com/github/SkyTNT/midi-model/blob/main/demo.ipynb)"

midi_model.py

@@ -125,7 +125,7 @@ class MIDIModel(nn.Module):
                 prompt = np.repeat(prompt, repeats=batch_size, axis=0)
             elif prompt.shape[0] == 1:
                 prompt = np.repeat(prompt, repeats=batch_size, axis=0)
-            else:
+            elif len(prompt.shape) != 3 or prompt.shape[0] != batch_size:
                 raise ValueError(f"invalid shape for prompt, {prompt.shape}")
             prompt = prompt[..., :max_token_seq]
             if prompt.shape[-1] < max_token_seq: