dar-tau commited on
Commit
b3fa350
·
verified ·
1 Parent(s): 436dc7d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -9
app.py CHANGED
@@ -169,18 +169,20 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
169
  # Here are some examples of prompts we can analyze their internal representations:
170
  # ''')
171
 
172
- # for info in dataset_info:
173
- # with gr.Tab(info['name']):
174
- # num_examples = 10
175
- # dataset = load_dataset(info['hf_repo'], split='train', streaming=True)
176
- # if 'filter' in info:
177
- # dataset = dataset.filter(info['filter'])
178
- # dataset = dataset.shuffle(buffer_size=2000).take(num_examples)
179
- # dataset = [[row[info['text_col']]] for row in dataset]
180
- # gr.Examples(dataset, [original_prompt_raw])
181
  gr.Markdown('''
182
  ## The Prompt to Analyze
183
  ''')
 
 
 
 
 
 
 
 
 
 
184
  with gr.Group():
185
  original_prompt_raw.render()
186
  original_prompt_btn = gr.Button('Compute', variant='primary')
 
169
  # Here are some examples of prompts we can analyze their internal representations:
170
  # ''')
171
 
172
+
 
 
 
 
 
 
 
 
173
  gr.Markdown('''
174
  ## The Prompt to Analyze
175
  ''')
176
+ for info in dataset_info:
177
+ with gr.Tab(info['name']):
178
+ num_examples = 10
179
+ dataset = load_dataset(info['hf_repo'], split='train', streaming=True)
180
+ if 'filter' in info:
181
+ dataset = dataset.filter(info['filter'])
182
+ dataset = dataset.shuffle(buffer_size=2000).take(num_examples)
183
+ dataset = [[row[info['text_col']]] for row in dataset]
184
+ gr.Examples(dataset, [original_prompt_raw])
185
+
186
  with gr.Group():
187
  original_prompt_raw.render()
188
  original_prompt_btn = gr.Button('Compute', variant='primary')