Update app.py

app.py

@@ -40,7 +40,6 @@ suggested_interpretation_prompts = [
 def initialize_gpu():
     pass
 
-
 def reset_model(model_name, *extra_components): 
     # extract model info
     model_args = deepcopy(model_info[model_name])
@@ -133,7 +132,7 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
             '''
                 **👾 This space is a simple introduction to the emerging trend of models interpreting their OWN hidden states in free form natural language!!👾**
                 This idea was investigated in the paper **Patchscopes** ([Ghandeharioun et al., 2024](https://arxiv.org/abs/2401.06102)) and was further explored in **SelfIE** ([Chen et al., 2024](https://arxiv.org/abs/2403.10949)). 
-                An honorary mention of **Speaking Probes** ([Dar, 2023](https://towardsdatascience.com/speaking-probes-self-interpreting-models-7a3dc6cb33d6) - my own work 🥳) which was less mature but had the same idea in mind. 
+                Honorary mention: **Speaking Probes** ([Dar, 2023](https://towardsdatascience.com/speaking-probes-self-interpreting-models-7a3dc6cb33d6) - my own work 🥳). It was less mature but had the same idea in mind. I think it's can be a great introduction to the subject!
                 We will follow the SelfIE implementation in this space for concreteness. Patchscopes are so general that they encompass many other interpretation techniques too!!! 
             ''', line_breaks=True)
             
@@ -204,10 +203,11 @@ 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), [interpretation_prompt, 
-                                                     num_tokens, do_sample, temperature, 
-                                                     top_k, top_p, repetition_penalty, length_penalty,
-                                                    ], [progress_dummy, *interpretation_bubbles])
+        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])
     
     original_prompt_btn.click(get_hidden_states, 
                               [original_prompt_raw],