Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -6,6 +6,9 @@ import time as reqtime
|
|
6 |
import datetime
|
7 |
from pytz import timezone
|
8 |
|
|
|
|
|
|
|
9 |
import numpy as np
|
10 |
|
11 |
import gradio as gr
|
@@ -216,21 +219,16 @@ if __name__ == "__main__":
|
|
216 |
print('Loading files list...')
|
217 |
|
218 |
all_MIDI_files_names = TMIDIX.Tegridy_Any_Pickle_File_Reader('all_MIDI_files_names')
|
219 |
-
|
220 |
-
print('Done!')
|
221 |
print('=' * 70)
|
222 |
|
223 |
print('Loading MIDI corpus embeddings...')
|
224 |
|
225 |
corpus_embeddings = np.load('MIDI_corpus_embeddings_all-mpnet-base-v2.npz')['data']
|
226 |
-
|
227 |
print('Done!')
|
228 |
print('=' * 70)
|
229 |
|
230 |
print('Loading Sentence Transformer model...')
|
231 |
-
|
232 |
model = SentenceTransformer('all-mpnet-base-v2')
|
233 |
-
|
234 |
print('Done!')
|
235 |
print('=' * 70)
|
236 |
|
|
|
6 |
import datetime
|
7 |
from pytz import timezone
|
8 |
|
9 |
+
from sentence_transformers import SentenceTransformer
|
10 |
+
from sentence_transformers import util
|
11 |
+
|
12 |
import numpy as np
|
13 |
|
14 |
import gradio as gr
|
|
|
219 |
print('Loading files list...')
|
220 |
|
221 |
all_MIDI_files_names = TMIDIX.Tegridy_Any_Pickle_File_Reader('all_MIDI_files_names')
|
|
|
|
|
222 |
print('=' * 70)
|
223 |
|
224 |
print('Loading MIDI corpus embeddings...')
|
225 |
|
226 |
corpus_embeddings = np.load('MIDI_corpus_embeddings_all-mpnet-base-v2.npz')['data']
|
|
|
227 |
print('Done!')
|
228 |
print('=' * 70)
|
229 |
|
230 |
print('Loading Sentence Transformer model...')
|
|
|
231 |
model = SentenceTransformer('all-mpnet-base-v2')
|
|
|
232 |
print('Done!')
|
233 |
print('=' * 70)
|
234 |
|