model_config.json

{
  "framework": "pytorch",
  "architecture": "efficientnet_b0_custom_head",
  "num_classes": 2,
  "image_size": 299,
  "class_indices": {
    "No-Stroke": 0,
    "Stroke": 1
  },
  "weights_file": "model.pth"
}