diff --git a/.github/docker-build-config.yml b/.github/docker-build-config.yml index c6930338..11bac0f9 100644 --- a/.github/docker-build-config.yml +++ b/.github/docker-build-config.yml @@ -1,6 +1,30 @@ -- backend_dir_name: segment_anything_model - backend_tag_prefix: sam- +- backend_dir_name: bert_classifier + backend_tag_prefix: bertclass- +- backend_dir_name: easyocr + backend_tag_prefix: easyocr- +- backend_dir_name: flair + backend_tag_prefix: flair- +- backend_dir_name: grounding_dino + backend_tag_prefix: grnddino- +- backend_dir_name: huggingface_llm + backend_tag_prefix: huggingfacellm- +- backend_dir_name: huggingface_ner + backend_tag_prefix: huggingfacener- +- backend_dir_name: interactive_substring_matching + backend_tag_prefix: interactsubstrmatch- +- backend_dir_name: langchain_search_agent + backend_tag_prefix: langchainsearch- - backend_dir_name: llm_interactive backend_tag_prefix: llm- -- backend_dir_name: the_simplest_backend - backend_tag_prefix: simplebackend- \ No newline at end of file +- backend_dir_name: mmdetection-3 + backend_tag_prefix: mmdetection3- +- backend_dir_name: nemo_asr + backend_tag_prefix: nemoasr- +- backend_dir_name: segment_anything_model + backend_tag_prefix: sam- +- backend_dir_name: sklearn_text_classifier + backend_tag_prefix: sklearntxtclass- +- backend_dir_name: spacy + backend_tag_prefix: spacy- +- backend_dir_name: tesseract + backend_tag_prefix: tesseract- diff --git a/label_studio_ml/examples/bert_classifier/docker-compose.yml b/label_studio_ml/examples/bert_classifier/docker-compose.yml index 7cf53723..8659c7bf 100644 --- a/label_studio_ml/examples/bert_classifier/docker-compose.yml +++ b/label_studio_ml/examples/bert_classifier/docker-compose.yml @@ -3,7 +3,7 @@ version: "3.8" services: bert_classifier: container_name: bert_classifier - image: humansignal/ml-backend:v0 + image: heartexlabs/label-studio-ml-backend:bertclass-master build: context: . args: diff --git a/label_studio_ml/examples/easyocr/docker-compose.yml b/label_studio_ml/examples/easyocr/docker-compose.yml index b27fb0ed..6b760aac 100644 --- a/label_studio_ml/examples/easyocr/docker-compose.yml +++ b/label_studio_ml/examples/easyocr/docker-compose.yml @@ -3,7 +3,7 @@ version: "3.8" services: easyocr: container_name: easyocr - image: humansignal/ml-backend:v0 + image: heartexlabs/label-studio-ml-backend:easyocr-master build: context: . args: diff --git a/label_studio_ml/examples/flair/docker-compose.yml b/label_studio_ml/examples/flair/docker-compose.yml index 7265477f..8ad056d3 100644 --- a/label_studio_ml/examples/flair/docker-compose.yml +++ b/label_studio_ml/examples/flair/docker-compose.yml @@ -3,7 +3,7 @@ version: "3.8" services: flair: container_name: flair - image: humansignal/ml-backend:v0 + image: heartexlabs/label-studio-ml-backend:flair-master build: context: . args: diff --git a/label_studio_ml/examples/grounding_dino/docker-compose.yml b/label_studio_ml/examples/grounding_dino/docker-compose.yml index 2333cb5b..da207bec 100644 --- a/label_studio_ml/examples/grounding_dino/docker-compose.yml +++ b/label_studio_ml/examples/grounding_dino/docker-compose.yml @@ -3,6 +3,7 @@ version: "3.11" services: grounding_dino: container_name: grounding_dino + image: heartexlabs/label-studio-ml-backend:grnddino-master build: context: . args: diff --git a/label_studio_ml/examples/huggingface_llm/docker-compose.yml b/label_studio_ml/examples/huggingface_llm/docker-compose.yml index 70aa93fd..b3c55cd3 100644 --- a/label_studio_ml/examples/huggingface_llm/docker-compose.yml +++ b/label_studio_ml/examples/huggingface_llm/docker-compose.yml @@ -3,7 +3,7 @@ version: "3.8" services: huggingface_llm: container_name: huggingface_llm - image: humansignal/ml-backend:v0 + image: heartexlabs/label-studio-ml-backend:huggingfacellm-master build: context: . args: diff --git a/label_studio_ml/examples/huggingface_ner/docker-compose.yml b/label_studio_ml/examples/huggingface_ner/docker-compose.yml index 53e10943..fdca68da 100644 --- a/label_studio_ml/examples/huggingface_ner/docker-compose.yml +++ b/label_studio_ml/examples/huggingface_ner/docker-compose.yml @@ -3,7 +3,7 @@ version: "3.8" services: huggingface_ner: container_name: huggingface_ner - image: humansignal/ml-backend:v0 + image: heartexlabs/label-studio-ml-backend:huggingfacener-master build: context: . args: diff --git a/label_studio_ml/examples/interactive_substring_matching/docker-compose.yml b/label_studio_ml/examples/interactive_substring_matching/docker-compose.yml index 0734ea89..993954df 100644 --- a/label_studio_ml/examples/interactive_substring_matching/docker-compose.yml +++ b/label_studio_ml/examples/interactive_substring_matching/docker-compose.yml @@ -3,7 +3,7 @@ version: "3.8" services: interactive_substring_matching: container_name: interactive_substring_matching - image: humansignal/ml-backend:v0 + image: heartexlabs/label-studio-ml-backend:interactsubstrmatch-master build: context: . args: diff --git a/label_studio_ml/examples/langchain_search_agent/docker-compose.yml b/label_studio_ml/examples/langchain_search_agent/docker-compose.yml index 793f0a44..240f2dd1 100644 --- a/label_studio_ml/examples/langchain_search_agent/docker-compose.yml +++ b/label_studio_ml/examples/langchain_search_agent/docker-compose.yml @@ -3,7 +3,7 @@ version: "3.8" services: langchain_search_agent: container_name: langchain_search_agent - image: humansignal/langchain:v0 + image: heartexlabs/label-studio-ml-backend:langchainsearch-master build: context: . args: diff --git a/label_studio_ml/examples/llm_interactive/docker-compose.yml b/label_studio_ml/examples/llm_interactive/docker-compose.yml index dbc5c158..45a29cd8 100644 --- a/label_studio_ml/examples/llm_interactive/docker-compose.yml +++ b/label_studio_ml/examples/llm_interactive/docker-compose.yml @@ -3,7 +3,7 @@ version: "3.8" services: llm_interactive: container_name: llm_interactive - image: heartexlabs/label-studio-ml-backend:llm-latest + image: heartexlabs/label-studio-ml-backend:llm-master build: context: . args: diff --git a/label_studio_ml/examples/mmdetection-3/docker-compose.yml b/label_studio_ml/examples/mmdetection-3/docker-compose.yml index f6dc0483..bb090c00 100644 --- a/label_studio_ml/examples/mmdetection-3/docker-compose.yml +++ b/label_studio_ml/examples/mmdetection-3/docker-compose.yml @@ -3,7 +3,7 @@ version: "3.8" services: mmdetection-3: container_name: mmdetection-3 - image: humansignal/ml-backend-mmdetection-3:v0 + image: heartexlabs/label-studio-ml-backend:mmdetection3-master build: context: . args: diff --git a/label_studio_ml/examples/nemo_asr/docker-compose.yml b/label_studio_ml/examples/nemo_asr/docker-compose.yml index 21233476..1b514b12 100644 --- a/label_studio_ml/examples/nemo_asr/docker-compose.yml +++ b/label_studio_ml/examples/nemo_asr/docker-compose.yml @@ -3,7 +3,7 @@ version: "3.8" services: nemo_asr: container_name: nemo_asr - image: humansignal/ml-backend:v0 + image: heartexlabs/label-studio-ml-backend:nemoasr-master build: context: . args: diff --git a/label_studio_ml/examples/segment_anything_model/docker-compose.yml b/label_studio_ml/examples/segment_anything_model/docker-compose.yml index 68cd88ca..6045aa4b 100644 --- a/label_studio_ml/examples/segment_anything_model/docker-compose.yml +++ b/label_studio_ml/examples/segment_anything_model/docker-compose.yml @@ -3,7 +3,7 @@ version: "3.8" services: segment_anything_model: container_name: segment_anything_model - image: heartexlabs/label-studio-ml-backend:sam-latest + image: heartexlabs/label-studio-ml-backend:sam-master build: context: . shm_size: '4gb' diff --git a/label_studio_ml/examples/sklearn_text_classifier/docker-compose.yml b/label_studio_ml/examples/sklearn_text_classifier/docker-compose.yml index 3368d4c9..76415892 100644 --- a/label_studio_ml/examples/sklearn_text_classifier/docker-compose.yml +++ b/label_studio_ml/examples/sklearn_text_classifier/docker-compose.yml @@ -3,7 +3,7 @@ version: "3.8" services: sklearn_text_classifier: container_name: sklearn_text_classifier - image: humansignal/ml-backend:v0 + image: heartexlabs/label-studio-ml-backend:sklearntxtclass-master build: context: . args: diff --git a/label_studio_ml/examples/spacy/docker-compose.yml b/label_studio_ml/examples/spacy/docker-compose.yml index 3e950232..129842ed 100644 --- a/label_studio_ml/examples/spacy/docker-compose.yml +++ b/label_studio_ml/examples/spacy/docker-compose.yml @@ -3,6 +3,7 @@ version: "3.8" services: spacy: container_name: spacy + image: heartexlabs/label-studio-ml-backend:spacy-master build: . environment: # specify the spacy model to use diff --git a/label_studio_ml/examples/tesseract/docker-compose.yml b/label_studio_ml/examples/tesseract/docker-compose.yml index 5efe855d..05c9899f 100644 --- a/label_studio_ml/examples/tesseract/docker-compose.yml +++ b/label_studio_ml/examples/tesseract/docker-compose.yml @@ -4,6 +4,7 @@ services: tesseract: container_name: tesseract + image: heartexlabs/label-studio-ml-backend:tesseract-master build: context: . args: