Update app.py

app.py

@@ -170,7 +170,7 @@ def remove_duplicate_pitches(tokens):
                     seen.add(pair[0])
                     deduped.extend(pair)
 
-        high_tokens = [t for t in chord if t > 18815]
+        high_tokens = [t for t in chord if 18816 < t < 18819]
 
         result.extend([header] + deduped + high_tokens)
 
@@ -230,6 +230,8 @@ def load_midi(input_midi):
             dur_vel = (8 * dur) + velocity
     
             melody_chords.extend([pat_ptc+256, dur_vel+16768])
+
+    melody_chords = remove_duplicate_pitches(melody_chords)
         
     return melody_chords
 
@@ -252,6 +254,8 @@ def save_midi(tokens):
 
     song_f = []
 
+    tokens = remove_duplicate_pitches(tokens)
+
     for ss in tokens:
 
         if 0 <= ss < 256:
@@ -327,9 +331,15 @@ def generate_music(prime, num_gen_tokens, num_mem_tokens, num_gen_batches, model
             return_prime=False,
             verbose=False
         )
+
+    outputs = []
+
+    for o in out.list()
+        outputs.append(remove_duplicate_pitches(o))
+        
     print("Done!")
     print_sep()
-    return out.tolist()
+    return outputs
 
 def generate_music_and_state(input_midi, num_prime_tokens, num_gen_tokens, num_mem_tokens,
                              model_temperature, final_composition, generated_batches, block_lines):