config.json

{
  "model_type": "text-classification",
  "description": "A simple custom text classification model",
  "framework": "pytorch"
}