asigalov61 commited on
Commit
6fce636
1 Parent(s): ab6bc31

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -12,7 +12,7 @@ import gradio as gr
12
 
13
  from x_transformer_1_23_2 import *
14
  import random
15
- import statistics
16
  import tqdm
17
 
18
  from midi_to_colab_audio import midi_to_colab_audio
@@ -177,12 +177,12 @@ def ClassifyMIDI(input_midi):
177
  y = out.tolist()
178
 
179
  output = [l[0] for l in y]
180
- result = statistics.mode(output)
181
 
182
  results.append(result)
183
 
184
  all_results_labels = [classifier_labels[0][r-384] for r in results]
185
- final_result = statistics.mode(results)
186
 
187
  print('Done!')
188
  print('=' * 70)
 
12
 
13
  from x_transformer_1_23_2 import *
14
  import random
15
+ from statistics import mode
16
  import tqdm
17
 
18
  from midi_to_colab_audio import midi_to_colab_audio
 
177
  y = out.tolist()
178
 
179
  output = [l[0] for l in y]
180
+ result = mode(output)
181
 
182
  results.append(result)
183
 
184
  all_results_labels = [classifier_labels[0][r-384] for r in results]
185
+ final_result = mode(results)
186
 
187
  print('Done!')
188
  print('=' * 70)