Update app.py

app.py

@@ -265,7 +265,7 @@ def generate_music_and_state(input_midi, num_prime_tokens, num_gen_tokens, num_m
         midi_fname, midi_score = save_midi(preview_tokens + tokens, batch_number=i)
         plot_kwargs = {'plot_title': f'Batch # {i}', 'return_plt': True}
         if len(final_composition) > PREVIEW_LENGTH:
-            plot_kwargs['preview_length_in_notes'] = int(PREVIEW_LENGTH / 3)
+            plot_kwargs['preview_length_in_notes'] = len([t for t in preview_tokens if t > 256])
         TMIDIX.Tegridy_ms_SONG_to_MIDI_Converter(
             midi_score,
             output_signature='Godzilla Piano Transformer',