Update app.py

app.py

@@ -203,11 +203,10 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
     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), dict(intepretation_prompt_raw=interpretation_prompt, 
-                                                         max_new_tokens=num_tokens, do_sample=do_sample, temperature=temperature, 
-                                                         top_k=top_k, top_p=top_p, repetition_penalty=repetition_penalty, 
-                                                         length_penalty=length_penalty,
-                                                        ), [progress_dummy, *interpretation_bubbles])
+        btn.click(partial(run_interpretation, i=i), [interpretation_prompt, 
+                                                     num_tokens, do_sample, temperature, 
+                                                     top_k, top_p, repetition_penalty, length_penalty
+                                                    ], [progress_dummy, *interpretation_bubbles])
     
     original_prompt_btn.click(get_hidden_states, 
                               [original_prompt_raw],