Update README.md

README.md

@@ -11,7 +11,7 @@ pipeline_tag: token-classification
 
 ## Introduction
 
-BertChunker is a text chunker based on BERT with a classifier head to predict the start token of chunks (for use in RAG, etc). It was finetuned on [sentence-transformers/all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2). The whole training lasted for 10 minutes on a Nvidia P40 GPU on a 50 MB synthetized dataset.
+BertChunker is a text chunker based on BERT with a classifier head to predict the start token of chunks (for use in RAG, etc). It was finetuned on [nreimers/MiniLM-L6-H384-uncased](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2). The whole training lasted for 10 minutes on a Nvidia P40 GPU on a 50 MB synthetized dataset.
 
 This repo includes model checkpoint, BertChunker class definition file and all the other files needed.
 
@@ -25,7 +25,7 @@ from modeling_bertchunker import BertChunker
 
 # load bert tokenizer
 tokenizer = AutoTokenizer.from_pretrained(
-    "sentence-transformers/all-MiniLM-L6-v2",
+    "nreimers/MiniLM-L6-H384-uncased",
     padding_side="right",
     model_max_length=255,
     trust_remote_code=True,
@@ -33,7 +33,7 @@ tokenizer = AutoTokenizer.from_pretrained(
 
 # load MiniLM-L6-H384-uncased bert config
 config = AutoConfig.from_pretrained(
-    "sentence-transformers/all-MiniLM-L6-v2",
+    "nreimers/MiniLM-L6-H384-uncased",
     trust_remote_code=True,
 )