Update app.py

app.py

@@ -202,9 +202,6 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
     
     
     # event listeners
-    extra_components = [interpretation_prompt, original_prompt_raw, 
-                        original_prompt_btn, *interpretation_bubbles]
-    model_chooser.change(reset_model, [model_chooser, *extra_components], extra_components)
     
     for i, btn in enumerate(tokens_container):
         btn.click(partial(run_interpretation, i=i), [interpretation_prompt, 
@@ -216,4 +213,9 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
                               [original_prompt_raw], 
                               [progress_dummy, *tokens_container, *interpretation_bubbles])
     original_prompt_raw.change(lambda: [gr.Button(visible=False) for _ in range(MAX_PROMPT_TOKENS)], [], tokens_container)
+
+    extra_components = [interpretation_prompt, original_prompt_raw, 
+                        original_prompt_btn, *interpretation_bubbles]
+    model_chooser.change(reset_model, [model_chooser, *extra_components], extra_components)
+    
     demo.launch()