config.json

{
  "architectures": [
    "CustomModelForImageClassification"
  ],
  "image_size": 224,
  "model_type": "custom_model",
  "torch_dtype": "float32",
  "transformers_version": "4.41.2"
}