debug synth lead class

app.py

@@ -21,13 +21,13 @@ col1, col2 = st.columns(2)
 with col1:
     instrument = st.selectbox(
         'Which intrument do you want ?',
-        ('🎸 Bass', '🎺  Brass', '🪈 Flute', '🪕 Guitar', '🎹 Keyboard', '🔨 Mallet', 'Organ', '🎷 Reed', '🎻 String', '⚡ Synth lead', '🎤 Vocal')
+        ('🎸 Bass', '🎺 Brass', '🪈 Flute', '🪕 Guitar', '🎹 Keyboard', '🔨 Mallet', 'Organ', '🎷 Reed', '🎻 String', '⚡ Synth lead', '🎤 Vocal')
     )
     
 with col2:
     instrument_t = st.selectbox(
         'Which type intrument do you want ?',
-        ('📯 Acoustic', '🎙️  Electronic', '🎛️ Synthetic')
+        ('📯 Acoustic', '🎙️ Electronic', '🎛️ Synthetic')
     ) 
     
 with st.expander("Magical parameters 🪄"):

model.py

@@ -22,8 +22,9 @@ def load_model(device):
 model = load_model(device)
 
 def format(text):
-    text = text.split(' ')[-1]
-    return text.replace(" ", "").lower()
+    stems = text.split(' ')[1:]
+    stems = [stem.replace(" ", "").lower() for stem in stems]
+    return '_'.join(stems)
 
 def choice_to_tensor(choice: Sequence[str]) -> torch.Tensor:
     choice = '_'.join([format(i) for i in choice])