Update README.md

README.md

@@ -12,6 +12,30 @@ The model was trained using bands 1-12 of the Sentinel-2 satellites and using th
 
 The input shape of the model is 224, 224, 12. 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
+    
+import tensorflow as tf
+from transformers import TFViTModel
+
+
+model = tf.keras.models.load_model('Sentinel-2_ViT_Autoencoder_RGB', custom_objects={"TFViTModel": TFViTModel})
+```
+
+You can extract the embeddings removing the last layer using:
+```
+import tensorflow as tf
+
+model = tf.keras.Sequential()
+
+for layer in autoencoder.layers[:-1]: # just exclude last layer from copying
+    model.add(layer)    
+        
+```
+
+
 ## Intended uses & limitations
 
 The model was trained with images of 10 different cities in Colombia, however it may require fine tuning or retraining to learn from other contexts such as countries and other continents.