Skip to content

inference.vit.img2label.v1

{
"$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"
]
}
}
}
}
}