Update README.md

README.md

@@ -14,7 +14,7 @@ tags:
 # Model Card for MITLL/LADI-v2-classifier-small-reference
 LADI-v2-classifier-small-reference is based on [google/bit-50](https://huggingface.co/google/bit-50) and fine-tuned on the LADI v2_resized dataset. LADI-v2-classifier is trained to identify labels of interest to disaster response managers from aerial images.
 
-🔴 __IMPORTANT__ ❗🔴 This model is the 'reference' version of the model, which is trained on 80% of the 10,000 available images. It is provided to facilitate reproduction of our paper and is not intended to be used in deployment.
+🔴 __IMPORTANT__ ❗🔴 This model is the 'reference' version of the model, which is trained on 80% of the 10,000 available images. It is provided to facilitate reproduction of our paper and is not intended to be used in deployment. For deployment, see the [MITLL/LADI-v2-classifier-small](https://huggingface.co/MITLL/LADI-v2-classifier-small) and [MITLL/LADI-v2-classifier-large](https://huggingface.co/MITLL/LADI-v2-classifier-large) models, which are trained on the full LADI v2 dataset (all splits).
 
 ## Model Details
 
@@ -34,7 +34,7 @@ The model architecture is based on Google's bit-50 model and fine-tuned on the L
 - trees_damage
 - water_any
 
-This 'reference' model is trained on the training split, which contains 8,000 images from 2015-2022. It is provided for the purpose of reproducing the results from the paper. The 'deploy' model is trained on the training, validation, and test sets, and contains 10,000 images from 2015-2023. We recommend that anyone who wishes to use this model in production use the deploy model.
+This 'reference' model is trained on the training split, which contains 8,000 images from 2015-2022. It is provided for the purpose of reproducing the results from the paper. The 'deploy' model is trained on the training, validation, and test sets, and contains 10,000 images from 2015-2023. We recommend that anyone who wishes to use this model in production use the main versions of the models [MITLL/LADI-v2-classifier-small](https://huggingface.co/MITLL/LADI-v2-classifier-small) and [MITLL/LADI-v2-classifier-large](https://huggingface.co/MITLL/LADI-v2-classifier-large).
 
 - **Developed by:** Jeff Liu, Sam Scheele
 - **Funded by:** Department of the Air Force under Air Force Contract No. FA8702-15-D-0001