Update app.py

app.py

@@ -220,7 +220,6 @@ if __name__ == "__main__":
     print('Loading MidiCaps dataset...')
     
     mc_dataset = load_dataset("amaai-lab/MidiCaps")
-    # mc_fnames = [f['location'].split('/')[-1].split('.mid')[0] for f in mc_dataset['train']]
     print('=' * 70)
     
     print('Loading files list...')
@@ -231,12 +230,12 @@ if __name__ == "__main__":
 
     print('Loading MIDI corpus embeddings...')
     
-    corpus_embeddings = np.load('MIDI_corpus_embeddings_all-MiniLM-L6-v2.npz')['data']
+    corpus_embeddings = np.load('MIDI_corpus_embeddings_all-mpnet-base-v2.npz')['data']
     print('Done!')
     print('=' * 70)
 
     print('Loading Sentence Transformer model...')
-    model = SentenceTransformer('all-MiniLM-L6-v2')
+    model = SentenceTransformer('all-mpnet-base-v2')
     print('Done!')
     print('=' * 70)