Update app.py
Browse files
app.py
CHANGED
@@ -186,7 +186,7 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
|
|
186 |
if 'filter' in info:
|
187 |
dataset = dataset.filter(info['filter'])
|
188 |
dataset = dataset.shuffle(buffer_size=2000).take(num_examples)
|
189 |
-
dataset = [[row[info['text_col']] + [gr.Button(visible=False) for _ in range(MAX_PROMPT_TOKENS)]
|
190 |
gr.Examples(dataset, [original_prompt_raw, *tokens_container], cache_examples=False)
|
191 |
|
192 |
with gr.Group():
|
|
|
186 |
if 'filter' in info:
|
187 |
dataset = dataset.filter(info['filter'])
|
188 |
dataset = dataset.shuffle(buffer_size=2000).take(num_examples)
|
189 |
+
dataset = [[row[info['text_col']]] + [gr.Button(visible=False) for _ in range(MAX_PROMPT_TOKENS)] for row in dataset]
|
190 |
gr.Examples(dataset, [original_prompt_raw, *tokens_container], cache_examples=False)
|
191 |
|
192 |
with gr.Group():
|