Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import note_seq
 import numpy as np
 from transformers import AutoTokenizer, AutoModelForCausalLM
+from constants import GM_INSTRUMENTS
 
 tokenizer = AutoTokenizer.from_pretrained("Katpeeler/lmd_8bars_tokenizer")
 model = AutoModelForCausalLM.from_pretrained("Katpeeler/midi_model_1")
@@ -10,7 +11,7 @@ NOTE_LENGTH_16TH_120BPM = 0.25 * 60 / 120
 BAR_LENGTH_120BPM = 4.0 * 60 / 120
 SAMPLE_RATE=44100
 
-def token_sequence_to_note_sequence(token_sequence, use_program=True, use_drums=True, instrument_mapper=None, only_piano=False):
+def token_sequence_to_note_sequence(token_sequence, use_program=True, use_drums=True, instrument_mapper=GM_INSTRUMENTS[], only_piano=False):
     if isinstance(token_sequence, str):
         token_sequence = token_sequence.split()
     note_sequence = empty_note_sequence()