From c369d5e51c57a5d0e17813dd9ab534c4a92f5b6e Mon Sep 17 00:00:00 2001 From: Jeff Vestal <53237856+jeffvestal@users.noreply.github.com> Date: Wed, 13 Dec 2023 09:27:01 -0600 Subject: [PATCH] Update load_embedding_model.ipynb --- load_embedding_model.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/load_embedding_model.ipynb b/load_embedding_model.ipynb index 95f72e1..571e2fd 100644 --- a/load_embedding_model.ipynb +++ b/load_embedding_model.ipynb @@ -166,7 +166,7 @@ "source": [ "# Set the model name from Hugging Face and task type\n", "hf_model_id='sentence-transformers/all-distilroberta-v1'\n", - "tm = TransformerModel(model_id=hf_model_id, task_type="text_embedding")\n", + "tm = TransformerModel(model_id=hf_model_id, task_type='text_embedding')\n", "\n", "#set the modelID as it is named in Elasticsearch\n", "es_model_id = tm.elasticsearch_model_id()\n",