Update app.py

app.py

@@ -345,6 +345,14 @@ def generate_callback_wrapper(input_midi,
     if not model_state:
         model_state = load_model(model_selector)
         model_state.append(model_selector)
+
+    else:
+        if model_selector != model_state[2]:
+            print('=' * 70)
+            print('Switching model...')
+            model_state = load_model(model_selector)
+            model_state.append(model_selector)
+            print('=' * 70)
     
     print('Num prime tokens:', num_prime_tokens)
     print('Num gen tokens:', num_gen_tokens)