Update app.py

app.py

@@ -2,7 +2,7 @@ from datasets import load_dataset
 from evaluate import evaluator
 from transformers import AutoModelForSequenceClassification, pipeline, AutoTokenizer
 import gradio as gr
-data = load_dataset("huolongguo10/check_sec_eval",split="train")
+data = load_dataset("huolongguo10/check_sec_eval",split="test")
 task_evaluator = evaluator("text-classification")
 model = AutoModelForSequenceClassification.from_pretrained("huolongguo10/check_sec")
 tokenizer = AutoTokenizer.from_pretrained("huolongguo10/check_sec")