Upload model_card.json with huggingface_hub

model_card.json

@@ -1,10 +1,10 @@
 {
-    "model_name": "attention_unet_model",
+    "model_name": "attention_unet_model.h5",
     "description": "This model performs brain tumor segmentation using an Attention UNet architecture.",
     "tasks": [
         "segmentation"
     ],
     "license": "apache-2.0",
     "dataset": "BraTS2020",
-    "model_architecture": "attention_unet_model"
+    "model_architecture": "attention_unet_model.h5"
 }