Update app.py

app.py

@@ -252,7 +252,7 @@ def generate_callback(input_midi,
     generated_batches = []
 
     if not final_composition and input_midi is not None:
-        final_composition = load_midi(input_midi)[:num_prime_tokens]
+        final_composition = load_midi(input_midi, model_selector=model_state[2])[:num_prime_tokens]
         midi_score = save_midi(final_composition, model_selector=model_state[2])
         block_lines.append(midi_score[-1][1] / 1000)