Update app.py

app.py

@@ -187,10 +187,7 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
                 dataset = dataset.filter(info['filter'])
             dataset = dataset.shuffle(buffer_size=2000).take(num_examples)
             dataset = [[row[info['text_col']]] for row in dataset]
-            gr.Examples(dataset, [original_prompt_raw], 
-                        # fn=lambda output: [[o[0]] + [gr.Button(visible=False) for _ in range(MAX_PROMPT_TOKENS)] for o in output],
-                        # run_on_click=True, 
-                        cache_examples=False)
+            gr.Examples(dataset, [original_prompt_raw], cache_examples=False)
             
     with gr.Group():
         original_prompt_raw.render()
@@ -238,5 +235,6 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
     
     original_prompt_btn.click(get_hidden_states, 
                               [original_prompt_raw], 
-                              [progress_dummy, global_state, *tokens_container, *interpretation_bubbles])    
+                              [progress_dummy, global_state, *tokens_container, *interpretation_bubbles])
+    original_prompt_raw.change(lambda: [gr.Button(visible=False) for _ in range(MAX_PROMPT_TOKENS)], [], tokens_container)
     demo.launch()