Update app.py

app.py

@@ -126,11 +126,12 @@ def load_midi(input_midi, model_selector=''):
     raw_score = TMIDIX.midi2single_track_ms_score(input_midi.name)
 
     escore_notes = TMIDIX.advanced_score_processor(raw_score, return_enhanced_score_notes=True, apply_sustain=True)[0]
-    escore_notes = TMIDIX.augment_enhanced_score_notes(escore_notes, timings_divider=32)
     
     sp_escore_notes = TMIDIX.solo_piano_escore_notes(escore_notes)
     zscore = TMIDIX.recalculate_score_timings(sp_escore_notes)
 
+    escore_notes = TMIDIX.augment_enhanced_score_notes(zscore, timings_divider=32)
+
     fscore = TMIDIX.fix_escore_notes_durations(zscore)
     
     cscore = TMIDIX.chordify_score([1000, fscore])