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",