huolongguo10 commited on
Commit
56776a9
1 Parent(s): 531f319

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -9,7 +9,8 @@ task_evaluator = evaluator("text-classification")
9
  eval_results = task_evaluator.compute(
10
  model_or_pipeline="huolongguo10/check_sec",
11
  data=data,
12
- input_column="sentence1"
 
13
  )
14
 
15
  with gr.Blocks() as demo:
 
9
  eval_results = task_evaluator.compute(
10
  model_or_pipeline="huolongguo10/check_sec",
11
  data=data,
12
+ input_column="sentence1",
13
+ label_mapping={"LABEL_0": 0, "LABEL_1": 1}
14
  )
15
 
16
  with gr.Blocks() as demo: