asigalov61 commited on
Commit
6c0dab8
1 Parent(s): 0e6d052

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ def find_midi(input_search_string):
52
  closest_index = np.argmax(similarities)
53
  closest_index_match_ratio = max(similarities[0].tolist())
54
 
55
- best_corpus_match = mc_dataset['train'][closest_index]
56
 
57
  print('Done!')
58
  print('=' * 70)
 
52
  closest_index = np.argmax(similarities)
53
  closest_index_match_ratio = max(similarities[0].tolist())
54
 
55
+ best_corpus_match = mc_dataset['train'][closest_index.tolist()]
56
 
57
  print('Done!')
58
  print('=' * 70)