Update README.md

README.md

@@ -24,7 +24,7 @@ metrics:
 # Stylizing ViT Small - Epithelium-Stroma *(Histopathology)*
 
 <!-- Provide a quick summary of what the model is/does. -->
-This model is the **Small** variant of **Stylizing ViT**, trained on the [**Aggregated Epithelium-Stroma**](https://github.com/chenxinli001/Task-Aug) (histopathology) dataset with the following splits: Train: {NKI} / Val: {VGH} / Test: {IHC}.
+This model is the **Small** variant of **Stylizing ViT**, trained on the [**Aggregated Epithelium-Stroma**](https://github.com/chenxinli001/Task-Aug) (histopathology) dataset with the following splits: **Train: {NKI} / Val: {VGH} / Test: {IHC}**.
 
 **Stylizing ViT** is a novel Vision Transformer encoder that utilizes weight-shared attention blocks for both self- and cross-attention. This design allows the same attention block to maintain anatomical consistency (via self-attention) while performing style transfer (via cross-attention), enabling anatomy-preserving instance style transfer for domain generalization in medical imaging.