config.json

{
  "model_type": "resnet-18",
  "architecture": "ResNet-18",
  "description": "ResNet-18 model trained to classify cats and dogs.",
  "framework": "pytorch",
  "task": "Image Classification",
  "tags": [
    "resnet",
    "image-classification",
    "cats",
    "dogs"
  ]
}