Update app.py

app.py

@@ -11,6 +11,8 @@ from sentence_transformers import util
 
 import numpy as np
 
+from datasets import load_dataset
+
 import gradio as gr
 
 import copy
@@ -223,17 +225,17 @@ if __name__ == "__main__":
     soundfont = "SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2"
     print('Loading files list...')
     
-    all_MIDI_files_names = TMIDIX.Tegridy_Any_Pickle_File_Reader('all_MIDI_files_names')
+    all_MIDI_files_names = TMIDIX.Tegridy_Any_Pickle_File_Reader('LAKH_all_MIDI_files_names')
     print('=' * 70)
 
     print('Loading MIDI corpus embeddings...')
     
-    corpus_embeddings = np.load('MIDI_corpus_embeddings_all-mpnet-base-v2.npz')['data']
+    corpus_embeddings = np.load('MIDI_corpus_embeddings_all-MiniLM-L6-v2.npz')['data']
     print('Done!')
     print('=' * 70)
 
     print('Loading Sentence Transformer model...')
-    model = SentenceTransformer('all-mpnet-base-v2')
+    model = SentenceTransformer('all-MiniLM-L6-v2')
     print('Done!')
     print('=' * 70)