Spaces:

huolongguo10
/

evaluator_cs

Sleeping

huolongguo10 commited on Apr 30, 2023

Commit

2ad9277

1 Parent(s): 9facccd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from transformers import AutoModelForSequenceClassification, pipeline, AutoToken
 import gradio as gr
 data = load_dataset("huolongguo10/insecure",split="train").shuffle(seed=42).select(range(1000))
 task_evaluator = evaluator("text-classification")
-model = AutoModelForSequenceClassification.from_pretrained("huolongguo10/check_sec").half()
 tokenizer = AutoTokenizer.from_pretrained("huolongguo10/check_sec")
 # 1. Pass a model name or path
 eval_results = task_evaluator.compute(

 import gradio as gr
 data = load_dataset("huolongguo10/insecure",split="train").shuffle(seed=42).select(range(1000))
 task_evaluator = evaluator("text-classification")
+model = AutoModelForSequenceClassification.from_pretrained("huolongguo10/check_sec")
 tokenizer = AutoTokenizer.from_pretrained("huolongguo10/check_sec")
 # 1. Pass a model name or path
 eval_results = task_evaluator.compute(