inference.vit.img2label.v2

Schema

{
  "$schema": "https://json-schema.org/draft/2020-12/schema",
  "type": "object",
  "required": [
    "type"
  ],
  "properties": {
    "type": {
      "type": "string",
      "enum": [
        "inference.vit.img2label.v1",
        "inference.vit.img2label.v2"
      ]
    },
    "config": {
      "type": "object",
      "properties": {
        "model": {
          "type": "string",
          "default": "Falconsai/nsfw_image_detection",
          "enum": [
            "google/vit-base-patch16-224-in21k",
            "Falconsai/nsfw_image_detection"
          ]
        }
      }
    }
  }
}