an0nymous commited on
Commit
41bc1b9
·
verified ·
1 Parent(s): 1e8383c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -20
app.py CHANGED
@@ -134,21 +134,22 @@ def plot_scatter_tab5(cat, x, y, z, col):
134
  return fig
135
 
136
 
137
- # # Tab 6
138
- # def random_sample(r: gr.Request, subset):
139
- # if subset is None or subset == []:
140
- # sample_index = np.random.randint(0, len(eval_set) - 1)
141
- # sample = eval_set[sample_index]
142
- # else: # filter by subsets (can be list)
143
- # if isinstance(subset, str):
144
- # subset = [subset]
145
- # # filter down dataset to only include the subset(s)
146
- # eval_set_filtered = eval_set.filter(lambda x: x["subset"] in subset)
147
- # sample_index = np.random.randint(0, len(eval_set_filtered) - 1)
148
- # sample = eval_set_filtered[sample_index]
149
-
150
- # markdown_text = '\n\n'.join([f"**{key}**:\n\n{value}" for key, value in sample.items()])
151
- # return markdown_text
 
152
 
153
  # Gradio Interface
154
  with gr.Blocks() as demo:
@@ -238,15 +239,14 @@ with gr.Blocks() as demo:
238
  with gr.TabItem("Dataset Viewer"):
239
  with gr.Row():
240
  # loads one sample
241
- gr.Markdown(
242
- """## Random Dataset Sample Viewer
243
- Warning, refusals, XSTest, and donotanswer datasets have sensitive content.""")
244
- subset_selector = gr.Dropdown(sub_cats, label="Subset", value=None, multiselect=True)
245
- button = gr.Button("Show Random Sample")
246
 
247
  with gr.Row():
248
  sample_display = gr.Markdown("{sampled data loads here}")
249
 
 
 
250
 
251
  # Launch the Gradio app
252
  demo.launch(share=True)
 
134
  return fig
135
 
136
 
137
+ # Tab 6
138
+ eval_set=raw_data
139
+ def random_sample(r: gr.Request, subset):
140
+ if subset is None or subset == []:
141
+ sample_index = np.random.randint(0, len(eval_set) - 1)
142
+ sample = eval_set[sample_index]
143
+ else: # filter by subsets (can be list)
144
+ if isinstance(subset, str):
145
+ subset = [subset]
146
+ # filter down dataset to only include the subset(s)
147
+ eval_set_filtered = eval_set.filter(lambda x: x["subset"] in subset)
148
+ sample_index = np.random.randint(0, len(eval_set_filtered) - 1)
149
+ sample = eval_set_filtered[sample_index]
150
+
151
+ markdown_text = '\n\n'.join([f"**{key}**:\n\n{value}" for key, value in sample.items()])
152
+ return markdown_text
153
 
154
  # Gradio Interface
155
  with gr.Blocks() as demo:
 
239
  with gr.TabItem("Dataset Viewer"):
240
  with gr.Row():
241
  # loads one sample
242
+ subset_selector = gr.Dropdown(sub_cats, label="Subset", value=None, multiselect=True)
243
+ button = gr.Button("Show Random Sample")
 
 
 
244
 
245
  with gr.Row():
246
  sample_display = gr.Markdown("{sampled data loads here}")
247
 
248
+ button.click(fn=random_sample, inputs=[subset_selector], outputs=[sample_display])
249
+
250
 
251
  # Launch the Gradio app
252
  demo.launch(share=True)