max-toxic-comment-classifier.yaml

# Copyright 2021 The MLX Contributors
#
# SPDX-License-Identifier: Apache-2.0

name: "Toxic Comment Classifier"
model_identifier: max-toxic-comment-classifier
description: "Detects 6 types of toxicity in user comments"
framework:
  name: pytorch
  version: "1.0.1"
license: "Apache 2.0"
domain: "Text Classification"
website: https://github.com/IBM/MAX-Toxic-Comment-Classifier/

serve:
  servable: true
  tested_platforms:
    - kubernetes
    - kfserving
  serving_container_image:
    container_image_url: "quay.io/codait/max-toxic-comment-classifier:latest"
    container_port: "5000"

readme_url: https://raw.githubusercontent.com/machine-learning-exchange/katalog/main/model-samples/max-toxic-comment-classifier/max-toxic-comment-classifier.md