Update app.py

app.py

@@ -12,6 +12,7 @@ import gradio as gr
 
 from x_transformer_1_23_2 import *
 import random
+import statistics
 import tqdm
 
 from midi_to_colab_audio import midi_to_colab_audio
@@ -197,8 +198,6 @@ def ClassifyMIDI(input_midi):
     for i, a in enumerate(all_results_labels):
       print('Notes', i*170, '-', (i*170)+340, '===', a)
     
-    print('=' * 70)
-        
     print('=' * 70)
     print('Done!')
     print('=' * 70)
@@ -271,11 +270,13 @@ if __name__ == "__main__":
     soundfont = "SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2"
 
     print('Loading Annotated MIDI Dataset processed scores...')
-    seed_melodies_data = TMIDIX.Tegridy_Any_Pickle_File_Reader('processed_scores')
+    processed_scores = TMIDIX.Tegridy_Any_Pickle_File_Reader('processed_scores')
+    print(processed_scores[0][0])
     print('=' * 70)
 
     print('Loading Annotated MIDI Dataset Classifier Songs Artists Labels...')
-    seed_melodies_data = TMIDIX.Tegridy_Any_Pickle_File_Reader('Annotated_MIDI_Dataset_Classifier_Songs_Artists_Labels')
+    classifier_labels = TMIDIX.Tegridy_Any_Pickle_File_Reader('Annotated_MIDI_Dataset_Classifier_Songs_Artists_Labels')
+    print(classifier_labels[0])
     print('=' * 70)
 
     app = gr.Blocks()
@@ -287,7 +288,7 @@ if __name__ == "__main__":
         
         input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"])
 
-        run_btn = gr.Button("generate", variant="primary")
+        run_btn = gr.Button("classify", variant="primary")
 
         gr.Markdown("## Classification results")