diff --git a/model/__init__.py b/model/__init__.py index 60de553..bdb979b 100644 --- a/model/__init__.py +++ b/model/__init__.py @@ -4,8 +4,8 @@ from model.six_seeds import request_inference_from_nachet_6seeds request_function = { - "swin_endpoint": request_inference_from_swin, - "seed_detector": request_inference_from_seed_detector, + "swin-endpoint": request_inference_from_swin, + "seed-detector": request_inference_from_seed_detector, "test": request_inference_from_test, - "nachet_6seeds": request_inference_from_nachet_6seeds + "nachet-6seeds": request_inference_from_nachet_6seeds } diff --git a/pipelines/pipeline_template.yaml b/pipelines/pipeline_template.yaml index 687377a..06c748d 100644 --- a/pipelines/pipeline_template.yaml +++ b/pipelines/pipeline_template.yaml @@ -5,7 +5,6 @@ date: pipelines: - models: - model_name: pipeline_name: created_by: creation_date: