Update README.md

README.md

@@ -1,3 +1,23 @@
+---
+language: en
+license: mit
+tags:
+  - deep-learning
+  - cancer-detection
+  - histopathology
+  - tensorflow
+  - efficientnet
+  - vision-transformer
+  - ViT
+  - medical-imaging
+model_name: EfficientNetV2S & ViT-Hybrid for Histopathologic Cancer Detection
+library_name: tensorflow
+datasets:
+  - histopathologic-cancer-detection
+  - PatchCamelyon
+---
+
+
 # Histopathologic Cancer Detection - EfficientNetV2S & ViT-Hybrid
 This repository contains models for detecting metastatic cancer in histopathologic images.
 - **EfficientNetV2S**: A Baseline CNN-based model for local feature extraction.
@@ -22,12 +42,16 @@ from tensorflow.keras.models import load_model
 ```
 
 # Download EfficientNetV2S model
+```sh
 model_path = hf_hub_download(repo_id="MooseML/EfficientNet-Cancer-Detection", filename="efficientnet_cancer_model.h5")
 model = load_model(model_path)
+```
 
 # Download ViT-Hybrid model
+```sh
 model_path_vit = hf_hub_download(repo_id="MooseML/EfficientNet-Cancer-Detection", filename="ViT_hybrid_cancer_model.h5")
 model_vit = load_model(model_path_vit)
+```
 
 ## Github and Kaggle Links for Full Training Pipeline
 - Full Training Code: [GitHub Repository](https://github.com/MooseML/Histopathologic-Cancer-Detection)