polinaeterna commited on
Commit
e6292dd
·
1 Parent(s): 4d70818

increase max num examples

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -325,7 +325,7 @@ with gr.Blocks() as demo:
325
 
326
  gr.Markdown("## Run nvidia quality classifier")
327
  batch_size = gr.Slider(0, 64, 32, step=4, label="Inference batch size", info="(set this to smaller value if this space crashes.)")
328
- num_examples = gr.Slider(0, 1000, 500, step=10, label="Number of examples", info="Number of random examples to run quality classifier on")
329
  gr_check_btn = gr.Button("Check Dataset")
330
  progress_bar = gr.Label(show_label=False)
331
  plot = gr.BarPlot()
@@ -346,8 +346,8 @@ with gr.Blocks() as demo:
346
  ["HuggingFaceFW/fineweb-edu", "default", "train", "text", None, 16, 500],
347
  # ["fka/awesome-chatgpt-prompts", "default", "train", "prompt", 64, 200],
348
  # ["proj-persona/PersonaHub", "instruction", "train", "synthesized text", 32, 1000],
349
- ["argilla/FinePersonas-v0.1", "default", "train", "persona", None, 64, 1000],
350
- ["allenai/real-toxicity-prompts", "default", "train", "continuation", "text", 64, 1000],
351
  ],
352
  [dataset_name, subset_dropdown, split_dropdown, text_column_dropdown, nested_text_column_dropdown, batch_size, num_examples],
353
  [progress_bar, plot, df_low, df_medium, df_high, texts_df],
 
325
 
326
  gr.Markdown("## Run nvidia quality classifier")
327
  batch_size = gr.Slider(0, 64, 32, step=4, label="Inference batch size", info="(set this to smaller value if this space crashes.)")
328
+ num_examples = gr.Slider(0, 2000, 500, step=10, label="Number of examples", info="Number of random examples to run quality classifier on")
329
  gr_check_btn = gr.Button("Check Dataset")
330
  progress_bar = gr.Label(show_label=False)
331
  plot = gr.BarPlot()
 
346
  ["HuggingFaceFW/fineweb-edu", "default", "train", "text", None, 16, 500],
347
  # ["fka/awesome-chatgpt-prompts", "default", "train", "prompt", 64, 200],
348
  # ["proj-persona/PersonaHub", "instruction", "train", "synthesized text", 32, 1000],
349
+ ["argilla/FinePersonas-v0.1", "default", "train", "persona", None, 64, 2000],
350
+ ["allenai/real-toxicity-prompts", "default", "train", "continuation", "text", 64, 2000],
351
  ],
352
  [dataset_name, subset_dropdown, split_dropdown, text_column_dropdown, nested_text_column_dropdown, batch_size, num_examples],
353
  [progress_bar, plot, df_low, df_medium, df_high, texts_df],