Update app.py

app.py

@@ -139,7 +139,7 @@ print('=' * 70)
 
 #==================================================================================
 
-def load_midi(input_midi):
+def load_midi(input_midi, input_mixed_pooling):
 
     raw_score = TMIDIX.midi2single_track_ms_score(input_midi)
     
@@ -229,9 +229,13 @@ def cosine_similarity_numpy(src_array, trg_array):
 
 #==================================================================================
 
-def select_best_output(outputs, embeddings, src_embeddings, top_k=10):
+def select_best_output(outputs, embeddings, src_embeddings, input_mixed_pooling, top_k=10):
 
-    emb_sims = cosine_similarity_numpy(src_embeddings, embeddings)
+    if input_mixed_pooling:
+        emb_sims = cosine_similarity_numpy(src_embeddings, embeddings)
+
+    else:
+        emb_sims = cosine_similarity_numpy(src_embeddings[:, :2048], embeddings[:, :2048])
 
     sorted_emb_sims = sorted(emb_sims, reverse=True)
 
@@ -265,6 +269,7 @@ def Classify_MIDI_Genre(input_midi):
     fn1 = fn.split('.')[0]
     
     print('Input MIDI file name:', fn)
+    print('Use mixed embeddings pooling:', input_mixed_pooling)
     print('=' * 70)
 
     #===============================================================================
@@ -295,7 +300,7 @@ def Classify_MIDI_Genre(input_midi):
 
     #===============================================================================
 
-    result = select_best_output(midi_gas_ps, midi_gas_pse, src_emb)
+    result = select_best_output(midi_gas_ps, midi_gas_pse, src_emb, input_mixed_pooling)
 
     results_str = ''
 
@@ -455,6 +460,8 @@ with gr.Blocks() as demo:
     input_midi = gr.File(label="Input MIDI", 
                          file_types=[".midi", ".mid", ".kar"]
                         )
+
+    input_mixed_pooling = gr.Checkbox(value=False, label="Use mixed embeddings pooling")
     
     generate_btn = gr.Button("Classify", variant="primary")
 
@@ -467,7 +474,8 @@ with gr.Blocks() as demo:
     output_cls_results = gr.Textbox(label="MIDI classification results")
 
     generate_btn.click(Classify_MIDI_Genre, 
-                       [input_midi
+                       [input_midi,
+                        input_mixed_pooling
                        ], 
                        [output_title,
                         output_audio,
@@ -478,8 +486,8 @@ with gr.Blocks() as demo:
                       )
     
     gr.Examples(
-                [["Hotel California.mid"],
-                 ["Come To My Window.mid"]
+                [["Hotel California.mid", False],
+                 ["Come To My Window.mid", False]
                 ],
                 [input_midi
                 ],