Update train_script.py

train_script.py

@@ -17,15 +17,14 @@ logging.basicConfig(
 
 # 1. Load a model to finetune with 2. (Optional) model card data
 model = SentenceTransformer(
-    "jinaai/jina-embeddings-v2-base-en",
-    trust_remote_code=True,
+    "microsoft/mpnet-base",
     model_card_data=SentenceTransformerModelCardData(
         language="en",
         license="apache-2.0",
-        model_name="jina-embeddings-v2-base-en trained on Natural Questions pairs",
+        model_name="MPNet base trained on Natural Questions pairs",
     ),
 )
-model_name = "jina-v2-base-natural-questions"
+model_name = "mpnet-base-natural-questions-mnrl"
 
 # 3. Load a dataset to finetune on
 dataset = load_dataset("sentence-transformers/natural-questions", split="train")