Update app.py

app.py

@@ -186,8 +186,8 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
             if 'filter' in info:
                 dataset = dataset.filter(info['filter'])
             dataset = dataset.shuffle(buffer_size=2000).take(num_examples)
-            dataset = [[row[info['text_col']]] + [gr.Button(visible=False) for _ in range(MAX_PROMPT_TOKENS)] for row in dataset]
-            gr.Examples(dataset, [original_prompt_raw, *tokens_container], cache_examples=False)
+            dataset = [row[info['text_col']] for row in dataset]
+            gr.Examples(dataset, [original_prompt_raw], cache_examples=False)
             
     with gr.Group():
         original_prompt_raw.render()