Update app.py

app.py

@@ -10,7 +10,7 @@ from transformers import (
 
 model_dir = "wsdwJohn1231/my-bert-model"
 
-config = AutoConfig.from_pretrained(model_dir, num_labels=3, finetuning_task="text-classification")
+config = AutoConfig.from_pretrained(model_dir, num_labels=2, finetuning_task="text-classification")
 tokenizer = AutoTokenizer.from_pretrained(model_dir)
 model = AutoModelForSequenceClassification.from_pretrained(model_dir, config=config)