Update app.py

app.py

@@ -260,10 +260,9 @@ with gr.Blocks(theme=gr.themes.Default(), css=css) as demo:
 
     # event listeners
     for i, btn in enumerate(tokens_container):
-        btn.click(partial(run_interpretation, i=i), [global_state, interpretation_prompt, 
+        btn.click(partial(run_interpretation, i=i, use_gpu=use_gpu), [global_state, interpretation_prompt, 
                                                      num_tokens, do_sample, temperature, 
                                                      top_k, top_p, repetition_penalty, length_penalty,
-                                                     use_gpu
                                                     ], [progress_dummy, *interpretation_bubbles])
     
     original_prompt_btn.click(get_hidden_states,