Update app.py

app.py

@@ -188,7 +188,7 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
             dataset = dataset.shuffle(buffer_size=2000).take(num_examples)
             dataset = [row[info['text_col']] for row in dataset]
             gr.Examples(dataset, [original_prompt_raw, *tokens_container], 
-                        fn=lambda output: output + [gr.Button(visible=False) for _ in range(MAX_PROMPT_TOKENS)],
+                        fn=lambda output: [o + [gr.Button(visible=False) for _ in range(MAX_PROMPT_TOKENS)] for o in output],
                         run_on_click=True, cache_examples=False)
             
     with gr.Group():