config.json

{
  "architecture": "CatDogClassifier",
  "num_classes": 2,
  "input_size": [
    3,
    224,
    224
  ],
  "classes": [
    "cat",
    "dog"
  ],
  "framework": "pytorch",
  "model_type": "image-classification"
}