mimbres commited on
Commit
2f76bec
1 Parent(s): badf747

Update model_helper.py

Browse files
Files changed (1) hide show
  1. 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, '/content/',
157
  audio_info['track_name'], model.midi_output_inverse_vocab)
158
  t.stop(); t.print_elapsed_time("post processing");
159
- midifile = os.path.join('/content/model_output/', audio_info['track_name'] + '.mid')
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