Update README.md

README.md

@@ -11,6 +11,18 @@ The model was trained using bands RGB (2, 3 and 4) (Red, Green and Blue) of the
 
 The input shape of the model is 224, 224, 3. To extract features you should remove the last layer.
 
+The model can be read as (example in jupyer):
+"""
+!git lfs install
+!git clone https://huggingface.co/MITCriticalData/Sentinel-2_ViT_Autoencoder_RGB_full_Colombia_Dataset
+    
+import tensorflow as tf
+from transformers import TFViTModel
+
+
+model = tf.keras.models.load_model('Sentinel-2_ViT_Autoencoder_RGB_full_Colombia_Dataset', custom_objects={"TFViTModel": TFViTModel})
+"""
+
 ## Intended uses & limitations
 
 The model was trained with images of 81 different cities in Colombia, however it may require fine tuning or retraining to learn from other contexts such as countries and other continents.