config.json

{
  "model_name": "camie-tagger",
  "model_type": "pytorch",
  "models": [
    {
      "name": "initial",
      "path": "model/model_initial_only.pt",
      "description": "Initial version of the tagger model"
    },
    {
      "name": "initial_st",
      "path": "model_initial.safetensors",
      "description": "Initial safetensors"
    },
    {
      "name": "refined",
      "path": "model/model_refined.pt",
      "description": "Refined version of the tagger model with improved performance"
    },
    {
      "name": "refined_st",
      "path": "model_refined.safetensors",
      "description": "Refined safetensors"
    },
    {
      "name": "onnx",
      "path": "model_initial.onnx",
      "description": "Onnx model"
    }
  ],
  "default_model": "initial",
  "description": "A custom tagging model with initial and refined versions",
  "version": "1.0.0",
  "tags": ["tagger", "classification"],
  "license": "gpl-3.0"
}