Update app.py

app.py

@@ -12,7 +12,7 @@ from transformers import (
 model_dir = "google-bert/bert-base-uncased"
 
 config = AutoConfig.from_pretrained(model_dir, num_labels=3, finetuning_task="text-classification")
-config.id2label = {'LABEL_0': "正面", 'LABEL_1': "负面", 'LABEL_2': "中性"}
+# config.id2label = {'LABEL_0': "正面", 'LABEL_1': "负面", 'LABEL_2': "中性"}
 tokenizer = AutoTokenizer.from_pretrained(model_dir)
 model = AutoModelForSequenceClassification.from_pretrained(model_dir, config=config)