File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
supporting-blog-content/plagiarism-detection-with-elasticsearch Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 9191 " # Set the model name from Hugging Face and task type\n " ,
9292 " # open ai detector model - developed by open ai https://github.com/openai/gpt-2-output-dataset/tree/master/detector\n " ,
9393 " hf_model_id='roberta-base-openai-detector'\n " ,
94- " tm = TransformerModel(hf_model_id, \" text_classification\" )\n " ,
94+ " tm = TransformerModel(model_id= hf_model_id, task_type= \" text_classification\" )\n " ,
9595 " \n " ,
9696 " #set the modelID as it is named in Elasticsearch\n " ,
9797 " es_model_id = tm.elasticsearch_model_id()\n " ,
121121 " # Set the model name from Hugging Face and task type\n " ,
122122 " # sentence-transformers model\n " ,
123123 " hf_model_id='sentence-transformers/all-mpnet-base-v2'\n " ,
124- " tm = TransformerModel(hf_model_id, \" text_embedding\" )\n " ,
124+ " tm = TransformerModel(model_id= hf_model_id, task_type= \" text_embedding\" )\n " ,
125125 " \n " ,
126126 " #set the modelID as it is named in Elasticsearch\n " ,
127127 " es_model_id = tm.elasticsearch_model_id()\n " ,
You can’t perform that action at this time.
0 commit comments