Katpeeler commited on
Commit
6b40c3f
·
1 Parent(s): d9b4da0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -4,14 +4,14 @@ import numpy as np
4
  from transformers import AutoTokenizer, AutoModelForCausalLM
5
  from constants import GM_INSTRUMENTS
6
 
7
- tokenizer = AutoTokenizer.from_pretrained("Katpeeler/lmd_8bars_tokenizer")
8
- model = AutoModelForCausalLM.from_pretrained("Katpeeler/midi_model_1")
9
 
10
  NOTE_LENGTH_16TH_120BPM = 0.25 * 60 / 120
11
  BAR_LENGTH_120BPM = 4.0 * 60 / 120
12
  SAMPLE_RATE=44100
13
 
14
- def token_sequence_to_note_sequence(token_sequence, use_program=True, use_drums=False, instrument_mapper=None, only_piano=True):
15
  if isinstance(token_sequence, str):
16
  token_sequence = token_sequence.split()
17
  note_sequence = empty_note_sequence()
 
4
  from transformers import AutoTokenizer, AutoModelForCausalLM
5
  from constants import GM_INSTRUMENTS
6
 
7
+ tokenizer = AutoTokenizer.from_pretrained("TristanBehrens/js-fakes-4bars")
8
+ model = AutoModelForCausalLM.from_pretrained("TristanBehrens/js-fakes-4bars")
9
 
10
  NOTE_LENGTH_16TH_120BPM = 0.25 * 60 / 120
11
  BAR_LENGTH_120BPM = 4.0 * 60 / 120
12
  SAMPLE_RATE=44100
13
 
14
+ def token_sequence_to_note_sequence(token_sequence, use_program=True, use_drums=True, instrument_mapper=None, only_piano=False):
15
  if isinstance(token_sequence, str):
16
  token_sequence = token_sequence.split()
17
  note_sequence = empty_note_sequence()