component.yaml

# Copyright 2021 The MLX Contributors
#
# SPDX-License-Identifier: Apache-2.0
name: 'Deploy Model on Kubernetes'
description: |
  Deploy AI models using Kubernetes deployment.
metadata:
  annotations: {platform: 'OpenSource'}
inputs:
  - {name: model_serving_image,          description: 'Required. Model serving container image'}
  - {name: deployment_name,              description: 'Model Deployment Name', default: 'model-serving'}
  - {name: container_port,               description: 'Application port number of the model container', default: '5000'}
  - {name: cleanup,                      description: 'Cleanup previous model deployments', default: 'False'}
implementation:
  container:
    image: docker.io/aipipeline/k8s-model-deploy:latest
    command: ['python']
    args: [
      -u, kube_deployment.py,
      --model_serving_image, {inputValue: model_serving_image},
      --deployment_name, {inputValue: deployment_name},
      --container_port, {inputValue: container_port},
      --cleanup, {inputValue: cleanup}
    ]