Update README.md

README.md

@@ -1,6 +1,6 @@
 ---
 library_name: transformers
-tags: []
+license: mit
 ---
 
 # Model card for RAD-DINO
@@ -24,27 +24,28 @@ RAD-DINO is described in detail in [RAD-DINO: Exploring Scalable Medical Image E
 
 <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
 
-### Downstream use
+RAD-DINO is shared for research purposes only.
+It is **not meant to be used for clinical practice**.
+
+<!-- ### Downstream use -->
 
 <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
 
-RAD-DINO is a vision backbone that can be plugged to other models for downstream tasks.
+The model is a vision backbone that can be plugged to other models for downstream tasks.
 Some potential uses are:
 
 - Image classification, with a classifier trained on top of the `CLS` token
 - Image segmentation, with a decoder trained using the patch tokens
+- Clustering, using the image embeddings directly
 - Image retrieval, using nearest neighbors of the CLS token
 - Report generation, with a language model to decode text
 
 Fine-tuning RAD-DINO is typically not necessary to obtain good performance in downstream tasks.
 
-### Out-of-scope use
+<!-- ### Out-of-scope use -->
 
 <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
 
-This model is shared for research purposes only.
-It is not meant to be used for clinical practice.
-
 ## Bias, risks, and limitations
 
 <!-- This section is meant to convey both technical and sociotechnical limitations. -->
@@ -116,7 +117,8 @@ torch.Size([1, 768, 16, 16])
 <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
 
 We used images from five public, deidentified chest X-ray datasets to train this checkpoint of RAD-DINO.
-Images in the validation and test sets from [MAIRA-1](https://arxiv.org/abs/2311.13668) were excluded from the training set of RAD-DINO.
+Images in the validation and test sets used to train [MAIRA](https://arxiv.org/abs/2311.13668) were excluded from the training set of RAD-DINO.
+The list of image files used for training is available at [`./training_images.csv`](./training_images.csv).
 
 | Dataset   | Num. images |
 | --------- | ----------: |
@@ -218,4 +220,4 @@ We used [SimpleITK](https://simpleitk.org/) and [Pydicom](https://pydicom.github
 
 ## Model card contact
 
-Fernando Pérez-García ([`fperezgarcia@microsoft.com`](mailto:fperezgarcia@microsoft.com)).
+Fernando Pérez-García ([`fperezgarcia@microsoft.com`](mailto:fperezgarcia@microsoft.com)).