Update README.md

README.md

@@ -36,8 +36,6 @@ This Vision Transformer model has been fine-tuned on Solar Dynamics Observatory
 <b>The data used was provided courtesy of NASA/SDO and the AIA, EVE, and HMI science teams.
 The authors gratefully acknowledge the entire NASA Solar Dynamics Observatory Mission Team.</b><br>
 
-For the SDO team: this model is the first version for demonstration purposes. It is only trained on the SDO Gallery data atm and we're working on additional data. 
-
 We will include more technical details here soon. 
 
 ## Example Images
@@ -53,7 +51,7 @@ We will include more technical details here soon.
 ![](images/XRay_irradiated_Nuclei.png)
 ![Right-click on this link (not the picture seen above) use 'save as'](https://roosevelt.devron-systems.com/HF/P278_73668090728-A7_003_027_proj.tif)
 
-## Training data
+## ViT base training data
 The ViT model was pretrained on a dataset consisting of 14 million images and 21k classes ([ImageNet-21k](http://www.image-net.org/).
 More information on the base model used can be found here: (https://huggingface.co/google/vit-base-patch16-224-in21k)