Update app.py

app.py

@@ -190,10 +190,10 @@ with gr.Blocks(theme=gr.themes.Default(), css=css) as demo:
                                  for i in range(model.config.num_hidden_layers)]
         
         for i, btn in enumerate(tokens_container):
-            btn.click(partial(run_interpretation, i=i), [progress_dummy, global_state, interpretation_prompt, 
+            btn.click(partial(run_interpretation, i=i), [global_state, interpretation_prompt, 
                                                          num_tokens, do_sample, temperature, 
                                                          top_k, top_p, repetition_penalty, length_penalty
-                                                              ], [*interpretation_bubbles])
+                                                        ], [progress_dummy, *interpretation_bubbles])
         
     original_prompt_btn.click(get_hidden_states, 
                               [original_prompt_raw],