Update app.py

app.py

@@ -400,18 +400,18 @@ def CompareMIDIs(input_src_midi, input_trg_midi, input_sampling_resolution, inpu
     print('=' * 70)
     print('Giant Music Transformer MIDI Comparator')
     print('=' * 70)
-
-    input_src_tokens = src_tokens
-    input_trg_tokens = trg_tokens
-    
+  
     sampling_resolution = max(40, min(1000, input_sampling_resolution)) * 3
     sampling_overlap = max(0, min(500, input_sampling_overlap)) * 3
     
     comp_length = (min(len(input_src_tokens), len(input_trg_tokens)) // sampling_resolution) * sampling_resolution
+
+    input_src_tokens = src_tokens[:comp_length]
+    input_trg_tokens = trg_tokens[:comp_length]
     
     comp_cos_sims = []
     
-    for i in range(0, comp_length-sampling_resolution-sampling_overlap), max(1, sampling_resolution-sampling_overlap)):
+    for i in range(0, comp_length, max(1, sampling_resolution-sampling_overlap)):
     
       torch.cuda.empty_cache()
     
@@ -437,7 +437,6 @@ def CompareMIDIs(input_src_midi, input_trg_midi, input_sampling_resolution, inpu
       cache = out[2]
       trg_embedings = cache.layer_hiddens[-1]
     
-    
       cos_sim = pairwise.cosine_similarity([src_embedings.cpu().detach().numpy()[0].flatten()],
                                           [trg_embedings.cpu().detach().numpy()[0].flatten()]
                                           ).tolist()[0][0]