Update app.py

app.py

@@ -149,10 +149,10 @@ def GenerateMIDI(num_tok, idrums, iinstr):
 
     midi_data = TMIDIX.score2midi(output, text_encoding)
 
-    with open(f"Allegro-Music-Transformer-Music-Composition.mid", 'wb') as f:
+    with open(f"Allegro-Music-Transformer-Composition.mid", 'wb') as f:
         f.write(midi_data)
 
-    output_plot = TMIDIX.plot_ms_SONG(output, plot_title='Allegro-Music-Transformer-Music-Composition', return_plt=True)
+    output_plot = TMIDIX.plot_ms_SONG(output[2], plot_title='Allegro-Music-Transformer-Composition', return_plt=True)
 
     audio = midi_to_colab_audio(new_fn, 
                         soundfont_path="SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2",
@@ -169,7 +169,7 @@ def GenerateMIDI(num_tok, idrums, iinstr):
     print('-' * 70)
     print('Req execution time:', (time.time() - start_time), 'sec')
     
-    return output_plot, "Allegro-Music-Transformer-Music-Composition.mid", (16000, audio)
+    return output_plot, "Allegro-Music-Transformer-Composition.mid", (16000, audio)
 
 # =================================================================================================