Update app.py

app.py

@@ -263,16 +263,20 @@ def ClassifyMIDI(input_midi):
     all_artists_labels = []
     
     for i, res in enumerate(results):
-      result_toks = [res[0]-512, res[1]-512]
-      song_artist = song_artist_tokens_to_song_artist(result_toks)
-      gidx = genre_labels_fnames.index(song_artist)
-      genre = genre_labels[gidx][1]
-      classification_summary_string += 'Notes ' + str(i*samples_overlap) + '-' +  str((i*samples_overlap)+340) + ' === ' + str(genre) + ' --- ' + str(song_artist))
-    
-      artist_label = str_strip_artist(song_artist.split(' --- ')[1])
-    
-      all_artists_labels.append(artist_label)
-    
+        result_toks = [res[0]-512, res[1]-512]
+        song_artist = song_artist_tokens_to_song_artist(result_toks)
+        gidx = genre_labels_fnames.index(song_artist)
+        genre = genre_labels[gidx][1]
+        
+        print('Notes', i*samples_overlap, '-', (i*samples_overlap)+340, '===', genre, '---', song_artist)
+        classification_summary_string += 'Notes ' + str(i*samples_overlap) + ' - ' +  str((i*samples_overlap)+340) + ' === ' + str(genre) + ' --- ' + str(song_artist)) + '\n'
+        
+        artist_label = str_strip_artist(song_artist.split(' --- ')[1])
+        
+        all_artists_labels.append(artist_label)
+        
+    classification_summary_string += '=' * 70
+    classification_summary_string += '\n'    
     print('=' * 70)
     
     mode_artist_label = mode(all_artists_labels)
@@ -280,31 +284,8 @@ def ClassifyMIDI(input_midi):
     
     print('Aggregated artist classification label:', mode_artist_label)
     print('Aggregated artist classification label ratio:', mode_artist_label_count / len(all_artists_labels))
-    
-    print('=' * 70)
-    print('Done!')
-    print('=' * 70)
-
-
-
-
-
-
-
-
-
-
-    
-
-
-    
-    print('All classification labels summary:')
-    print('=' * 70)
-    
-    for i, a in enumerate(all_results_labels):
-        print('Notes', i*85, '-', (i*85)+340, '===', a)
-        classification_summary_string += 'Notes ' + str(i*85) + ' - ' + str((i*85)+340) + ' === ' + str(a) + '\n'
-    
+    classification_summary_string += 'Aggregated artist classification label: ' + str(mode_artist_label) + '\n'
+    classification_summary_string += 'Aggregated artist classification label ratio: ' + str(mode_artist_label_count / len(all_artists_labels)) + '\n'
     classification_summary_string += '=' * 70
     classification_summary_string += '\n'
     
@@ -312,18 +293,6 @@ def ClassifyMIDI(input_midi):
     print('Done!')
     print('=' * 70)
     
-    #===============================================================================
-
-    #========================================================
-
-    output_midi_title = str(fn1)
-    output_midi_summary = classification_summary_string
-    
-    print('Output MIDI file name:', mc_song_artist)
-    print('Output MIDI title:', output_midi_title)
-    print('=' * 70) 
-    
-
     #========================================================
     
     print('-' * 70)
@@ -397,12 +366,12 @@ if __name__ == "__main__":
 
     app = gr.Blocks()
     with app:
-        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Advanced MIDI Classifier</h1>")
-        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Detailed MIDI classification with transformers</h1>")
+        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Ultimate MIDI Classifier</h1>")
+        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Classify absolutely any MIDI by genre, song and artist</h1>")
         gr.Markdown(
-            "![Visitors](https://api.visitorbadge.io/api/visitors?path=asigalov61.Advanced-MIDI-Classifier&style=flat)\n\n"
-            "This is a demo for Annotated MIDI Dataset\n\n"
-            "Check out [Annotated MIDI Dataset](https://huggingface.co/datasets/asigalov61/Annotated-MIDI-Dataset) on Hugging Face!\n\n"
+            "![Visitors](https://api.visitorbadge.io/api/visitors?path=asigalov61.Ultimate-MIDI-Classifier&style=flat)\n\n"
+            "This is a demo for Ultimate MIDI Classifier\n\n"
+            "Check out [Ultimate MIDI Classifier](https://github.com/asigalov61/Ultimate-MIDI-Classifier) on GitHub!\n\n"
         )
 
         input_midi = gr.File(label="Input MIDI", file_types=[".midi", ".mid", ".kar"])