Update app.py

app.py

@@ -22,7 +22,7 @@ os.environ['USE_FLASH_ATTENTION'] = '1'
 
 import torch
 
-torch.set_float32_matmul_precision('medium')
+torch.set_float32_matmul_precision('high')
 torch.backends.cuda.matmul.allow_tf32 = True # allow tf32 on matmul
 torch.backends.cudnn.allow_tf32 = True # allow tf32 on cudnn
 torch.backends.cuda.enable_mem_efficient_sdp(True)
@@ -61,11 +61,9 @@ print('=' * 70)
 
 MODEL_CHECKPOINT = 'Guided_Accompaniment_Transformer_Trained_Model_36457_steps_0.5384_loss_0.8417_acc.pth'
 
-SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
+SOUNDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
 
-MAX_MELODY_NOTES = 64
-
-MAX_GEN_TOKS = 3072
+MAX_MELODY_NOTES = 128
 
 #==================================================================================
 
@@ -126,14 +124,14 @@ print('=' * 70)
 
 #==================================================================================
 
-def load_midi(input_midi, melody_patch=-1, use_nth_note=1):
+def load_midi(input_midi, melody_patch=-1):
 
     raw_score = TMIDIX.midi2single_track_ms_score(input_midi)
 
     escore_notes = TMIDIX.advanced_score_processor(raw_score, return_enhanced_score_notes=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)
 
     if melody_patch == -1:
         zscore = TMIDIX.recalculate_score_timings(sp_escore_notes)
@@ -147,31 +145,22 @@ def load_midi(input_midi, melody_patch=-1, use_nth_note=1):
         else:
             zscore = TMIDIX.recalculate_score_timings(sp_escore_notes)
     
-    cscore = TMIDIX.chordify_score([1000, zscore])[:MAX_MELODY_NOTES:use_nth_note]
+    cscore = TMIDIX.chordify_score([1000, zscore])[:MAX_MELODY_NOTES]
     
     score = []
     
-    score_list = []
-    
     pc = cscore[0]
     
     for c in cscore:
         score.append(max(0, min(127, c[0][1]-pc[0][1])))
-    
-        scl = [[max(0, min(127, c[0][1]-pc[0][1]))]]
-    
+
         n = c[0]
         
         score.extend([max(1, min(127, n[2]))+128, max(1, min(127, n[4]))+256])
-        scl.append([max(1, min(127, n[2]))+128, max(1, min(127, n[4]))+256])
-    
-        score_list.append(scl)
     
         pc = c
-    
-    score_list.append(scl)
 
-    return score, score_list
+    return score
 
 #==================================================================================
 
@@ -179,7 +168,6 @@ def load_midi(input_midi, melody_patch=-1, use_nth_note=1):
 def Generate_Accompaniment(input_midi,
                            input_melody,
                            melody_patch,
-                           use_nth_note,
                            model_temperature,
                            model_sampling_top_k
                           ):
@@ -250,7 +238,6 @@ def Generate_Accompaniment(input_midi,
     else:
         print('Input sample melody:', input_melody)
     print('Source melody patch:', melody_patch)
-    print('Use nth melody note:', use_nth_note)
     print('Model temperature:', model_temperature)
     print('Model top k:', model_sampling_top_k)
    
@@ -433,7 +420,6 @@ with gr.Blocks() as demo:
     gr.Markdown("## Generation options")
     
     melody_patch = gr.Slider(-1, 127, value=-1, step=1, label="Source melody MIDI patch")
-    use_nth_note = gr.Slider(1, 8, value=1, step=1, label="Use each nth melody note")
     model_temperature = gr.Slider(0.1, 1, value=0.9, step=0.01, label="Model temperature")
     model_sampling_top_k = gr.Slider(1, 100, value=15, step=1, label="Model sampling top k value")
     
@@ -450,7 +436,6 @@ with gr.Blocks() as demo:
                        [input_midi,
                         input_melody,
                         melody_patch,
-                        use_nth_note,
                         model_temperature,
                         model_sampling_top_k
                        ], 
@@ -462,13 +447,12 @@ with gr.Blocks() as demo:
                       )
 
     gr.Examples(
-                [["USSR-National-Anthem-Seed-Melody.mid", "Custom MIDI", -1, 1, 0.9, 15],
-                 ["Sparks-Fly-Seed-Melody.mid", "Custom MIDI", -1, 1, 0.9, 15]
+                [["USSR-National-Anthem-Seed-Melody.mid", "Custom MIDI", -1, 0.9, 15],
+                 ["Sparks-Fly-Seed-Melody.mid", "Custom MIDI", -1, 0.9, 15]
                 ],
                 [input_midi,
                  input_melody,
                  melody_patch,
-                 use_nth_note,
                  model_temperature,
                  model_sampling_top_k
                 ],