Update app.py

app.py

@@ -12,7 +12,7 @@ model_dir = "my-bert-model"
 
 config = AutoConfig.from_pretrained(model_dir, num_labels=3, finetuning_task="text-classification")
 tokenizer = AutoTokenizer.from_pretrained(model_dir)
-model = AutoModelForSequenceClassification.from_pretrained(model_dir, config=config)
+model = AutoModelForSequenceClassification.from_pretrained(model_dir, config=config, ignore_mismatched_sizes=True)
 
 def inference(input_text):
     inputs = tokenizer.batch_encode_plus(