Update README.md

README.md

@@ -4,6 +4,8 @@ library_name: stylizing-vit
 tags:
 - style-transfer
 - medical
+- laparoscopy
+- cholecystectomy
 - domain-generalization
 - vision-transformer
 - pytorch
@@ -20,10 +22,10 @@ metrics:
 - Accuracy
 ---
 
-# Stylizing ViT Base - Cholec80
+# Stylizing ViT Base - Cholec80 *(Laparoscopy, Cholecystectomy)*
 
 <!-- Provide a quick summary of what the model is/does. -->
-This model is the **Base** variant of **Stylizing ViT**, trained on the [**Cholec80**]([[DATASET_LINK]](https://zenodo.org/records/13170928)) dataset. 
+This model is the **Base** variant of **Stylizing ViT**, trained on the [**Cholec80**]([[DATASET_LINK]](https://zenodo.org/records/13170928)) (laparoscopy, cholecystectomy) dataset with the following splits: **Train: {41, 42} / Val: {43} / Test: {44, 45}**.
 
 **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.