Spaces:
Sleeping
Sleeping
Commit
·
9aa421e
1
Parent(s):
1c0a85d
Update app.py
Browse files
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").
|
|
|
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:
|