kaycebasques
diff --git a/‎generative_ai/embeddings/batch_example.py
Copy file name to clipboardExpand all lines: generative_ai/embeddings/batch_example.py
+1-1Lines changed: 1 addition & 1 deletion b/‎generative_ai/embeddings/batch_example.py
Copy file name to clipboardExpand all lines: generative_ai/embeddings/batch_example.py
+1-1Lines changed: 1 addition & 1 deletion
@@ -39,7 +39,7 @@ def embed_text_batch() -> BatchPredictionJob:
     output_uri = OUTPUT_URI
 
     textembedding_model = language_models.TextEmbeddingModel.from_pretrained(
-        "gemini-embedding-001"
+        "textembedding-gecko@003"
     )
 
     batch_prediction_job = textembedding_model.batch_predict(
Original file line number	Diff line number	Diff line change
`@@ -39,7 +39,7 @@ def embed_text_batch() -> BatchPredictionJob:`
`39`	`39`	`output_uri = OUTPUT_URI`
`40`	`40`
`41`	`41`	`textembedding_model = language_models.TextEmbeddingModel.from_pretrained(`
`42`		`- "gemini-embedding-001"`
	`42`	`+ "textembedding-gecko@003"`
`43`	`43`	`)`
`44`	`44`
`45`	`45`	`batch_prediction_job = textembedding_model.batch_predict(`