dar-tau commited on
Commit
f284d65
·
verified ·
1 Parent(s): d028e6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)]] for row in dataset]
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():