add

source/services/ner/train/train.py

@@ -346,7 +346,7 @@ unwrapped_model.save_pretrained(output_dir, save_function=accelerator.save)
 from transformers import pipeline
 
 # Replace this with your own checkpoint
-model_checkpoint = "aimlnerd/bert-finetuned-legalentity-ner"
+model_checkpoint = "aimlnerd/bert-finetuned-legalentity-ner-accelerate"
 token_classifier = pipeline(
     "token-classification", model=model_checkpoint, aggregation_strategy="simple"
 )