Update README.md

README.md

@@ -26,7 +26,7 @@ WhisperNER supports open-type NER, enabling recognition of diverse and evolving
 ---------
 
 ## Training Details
-`aiola/whisper-ner-v1` was trained on the NuNER dataset to perform joint audio transcription and NER tagging. 
+`aiola/whisper-ner-tag-and-mask-v1` was finetuned from `aiola/whisper-ner-v1` using the NuNER dataset to perform joint audio transcription and NER tagging or NER masking. 
 The model was trained and evaluated only on English data. Check out the [paper](https://arxiv.org/abs/2409.08107) for full details.
 
 ---------
@@ -39,7 +39,7 @@ Inference can be done using the following code (for inference code and more deta
 import torch
 from transformers import WhisperProcessor, WhisperForConditionalGeneration
 
-model_path = "aiola/whisper-ner-v1"
+model_path = "aiola/whisper-ner-tag-and-mask-v1"
 audio_file_path = "path/to/audio/file"
 prompt = "person, company, location"  # comma separated entity tags
 apply_entity_mask = False  # change to True for entity masking