huolongguo10 commited on
Commit
9facccd
1 Parent(s): 9aa421e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  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",split="train").shuffle(seed=42).select(range(1000))
6
  task_evaluator = evaluator("text-classification")
 
1
  from datasets import load_dataset
2
  from evaluate import evaluator
3
+ from transformers import AutoModelForSequenceClassification, pipeline, AutoTokenizer
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")