config.json

{
    "architectures": [
        "TechSupportClassifier"
    ],
    "model_type": "custom",
    "num_category_labels": 5,
    "num_urgency_labels": 3,
    "auto_map": {
        "AutoModel": "modeling_tech_support.TechSupportClassifier"
    }
}