asigalov61 commited on
Commit
325e435
1 Parent(s): 340485a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
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