config.json

{
  "model_name": "microsoft/resnet-50",
  "num_classes": 2
}