From c8fedbb40786330cb7d8131de11c75cc85eb6010 Mon Sep 17 00:00:00 2001
From: Jeff Vestal <53237856+jeffvestal@users.noreply.github.com>
Date: Wed, 13 Dec 2023 09:26:13 -0600
Subject: [PATCH] updated TransformerModel init

tm = TransformerModel(model_id=hf_model_id, task_type="text_embedding")
---
 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 6120b52..95f72e1 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(hf_model_id, \"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",