asigalov61 commited on
Commit
d36ebf1
·
verified ·
1 Parent(s): c26a29c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -7
app.py CHANGED
@@ -32,8 +32,9 @@ def TranscribePianoAudio(input_file):
32
  start_time = reqtime.time()
33
  print('=' * 70)
34
 
35
- fn = str(input_file)
36
  fn1 = fn.split('.')[0]
 
37
 
38
  print('-' * 70)
39
  print('Input file name:', fn)
@@ -41,7 +42,7 @@ def TranscribePianoAudio(input_file):
41
  print('Loading audio...')
42
 
43
  # Load audio
44
- (audio, _) = load_audio(input_file, sr=sample_rate, mono=True)
45
  print('Done!')
46
  print('-' * 70)
47
  print('Loading transcriptor..')
@@ -52,12 +53,12 @@ def TranscribePianoAudio(input_file):
52
  print('-' * 70)
53
  print('Transcribing...')
54
 
55
- transcribed_dict = transcriptor.transcribe(audio, fn)
56
  print('Done!')
57
  print('-' * 70)
58
 
59
  #===============================================================================
60
- raw_score = TMIDIX.midi2single_track_ms_score(fn)
61
 
62
  #===============================================================================
63
  # Enhanced score notes
@@ -78,7 +79,7 @@ def TranscribePianoAudio(input_file):
78
 
79
  print('=' * 70)
80
 
81
- audio = midi_to_colab_audio(fn,
82
  soundfont_path=soundfont,
83
  sample_rate=16000,
84
  volume_scale=10,
@@ -92,10 +93,10 @@ def TranscribePianoAudio(input_file):
92
 
93
  output_midi_title = str(fn1)
94
  output_midi_summary = str(escore[:3])
95
- output_midi = str(fn)
96
  output_audio = (16000, audio)
97
 
98
- output_plot = TMIDIX.plot_ms_SONG(escore, plot_title=output_midi, return_plt=True)
99
 
100
  print('Output MIDI file name:', output_midi)
101
  print('Output MIDI title:', output_midi_title)
 
32
  start_time = reqtime.time()
33
  print('=' * 70)
34
 
35
+ fn = os.path.basename(input_file)
36
  fn1 = fn.split('.')[0]
37
+ out_mid = fn1+'.mid'
38
 
39
  print('-' * 70)
40
  print('Input file name:', fn)
 
42
  print('Loading audio...')
43
 
44
  # Load audio
45
+ (audio, _) = load_audio(fn, sr=sample_rate, mono=True)
46
  print('Done!')
47
  print('-' * 70)
48
  print('Loading transcriptor..')
 
53
  print('-' * 70)
54
  print('Transcribing...')
55
 
56
+ transcribed_dict = transcriptor.transcribe(audio, out_mid)
57
  print('Done!')
58
  print('-' * 70)
59
 
60
  #===============================================================================
61
+ raw_score = TMIDIX.midi2single_track_ms_score(out_mid)
62
 
63
  #===============================================================================
64
  # Enhanced score notes
 
79
 
80
  print('=' * 70)
81
 
82
+ audio = midi_to_colab_audio(out_mid,
83
  soundfont_path=soundfont,
84
  sample_rate=16000,
85
  volume_scale=10,
 
93
 
94
  output_midi_title = str(fn1)
95
  output_midi_summary = str(escore[:3])
96
+ output_midi = str(out_mid)
97
  output_audio = (16000, audio)
98
 
99
+ output_plot = TMIDIX.plot_ms_SONG(escore, plot_title=output_midi_title, return_plt=True)
100
 
101
  print('Output MIDI file name:', output_midi)
102
  print('Output MIDI title:', output_midi_title)