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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -4,13 +4,15 @@ 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
- model = AutoModelForSequenceClassification.from_pretrained("huolongguo10/check_sec").float()
 
8
  # 1. Pass a model name or path
9
  eval_results = task_evaluator.compute(
10
  model_or_pipeline=model,
11
  data=data,
12
  input_column="sentence1",
13
- label_mapping={"LABEL_0": 0, "LABEL_1": 1}
 
14
  )
15
 
16
  with gr.Blocks() as demo:
 
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
+ model = AutoModelForSequenceClassification.from_pretrained("huolongguo10/check_sec").half()
8
+ tokenizer = AutoTokenizer.from_pretrained("huolongguo10/check_sec")
9
  # 1. Pass a model name or path
10
  eval_results = task_evaluator.compute(
11
  model_or_pipeline=model,
12
  data=data,
13
  input_column="sentence1",
14
+ label_mapping={"LABEL_0": 0, "LABEL_1": 1},
15
+ tokenizer=tokenizer
16
  )
17
 
18
  with gr.Blocks() as demo: