Update app.py

app.py

@@ -123,13 +123,15 @@ 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)[0]
+    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, keep_drums=False)
+    sp_escore_notes = TMIDIX.solo_piano_escore_notes(escore_notes)
     zscore = TMIDIX.recalculate_score_timings(sp_escore_notes)
+
+    fscore = TMIDIX.fix_escore_notes_durations(zscore)
     
-    cscore = TMIDIX.chordify_score([1000, zscore])
+    cscore = TMIDIX.chordify_score([1000, fscore])
     
     score = []