Update app.py

app.py

@@ -19,7 +19,7 @@ model = BertForSequenceClassification(config)
 # model.load_state_dict(torch.load(os.path.join(bert_out_address, "pytorch_model.bin")))
 
 model.load_state_dict(torch.load(os.path.join(bert_out_address, "pytorch_model.bin"), map_location=torch.device(device)))
-model = BertForTokenClassification.from_pretrained(bert_out_address,num_labels=3)
+model = BertForTokenClassification.from_pretrained(bert_out_address,num_labels=5)
 
 # Load the tokenizer
 tokenizer = BertTokenizer.from_pretrained(bert_out_address)