huolongguo10 commited on
Commit
3eacfc7
1 Parent(s): 6866515

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ from datasets import load_dataset
2
  from evaluate import evaluator
3
  from transformers import AutoModelForSequenceClassification, pipeline
4
  import gradio as gr
5
- data = load_dataset("huolongguo10/insecure")
6
  task_evaluator = evaluator("text-classification")
7
 
8
  # 1. Pass a model name or path
 
2
  from evaluate import evaluator
3
  from transformers import AutoModelForSequenceClassification, pipeline
4
  import gradio as gr
5
+ data = load_dataset("huolongguo10/insecure",split="train").shuffle(seed=42).select(range(1000))
6
  task_evaluator = evaluator("text-classification")
7
 
8
  # 1. Pass a model name or path