Spaces:
Sleeping
Sleeping
Update model_helper.py
Browse files- model_helper.py +2 -2
model_helper.py
CHANGED
@@ -153,9 +153,9 @@ def transcribe(model, audio_info):
|
|
153 |
pred_notes = mix_notes(pred_notes_in_file) # This is the mixed notes from all channels
|
154 |
|
155 |
# Write MIDI
|
156 |
-
write_model_output_as_midi(pred_notes, '
|
157 |
audio_info['track_name'], model.midi_output_inverse_vocab)
|
158 |
t.stop(); t.print_elapsed_time("post processing");
|
159 |
-
midifile = os.path.join('
|
160 |
assert os.path.exists(midifile)
|
161 |
return midifile
|
|
|
153 |
pred_notes = mix_notes(pred_notes_in_file) # This is the mixed notes from all channels
|
154 |
|
155 |
# Write MIDI
|
156 |
+
write_model_output_as_midi(pred_notes, './',
|
157 |
audio_info['track_name'], model.midi_output_inverse_vocab)
|
158 |
t.stop(); t.print_elapsed_time("post processing");
|
159 |
+
midifile = os.path.join('./model_output/', audio_info['track_name'] + '.mid')
|
160 |
assert os.path.exists(midifile)
|
161 |
return midifile
|