Update README

README.md

@@ -8,6 +8,79 @@ base_model:
 - laion/CLIP-ViT-B-16-laion2B-s34B-b88K
 ---
 
+[![arXiv](https://img.shields.io/badge/arXiv-2504.11171-b31b1b?logo=arxiv)](https://arxiv.org/abs/2503.15969)
+[![Code](https://img.shields.io/badge/GitHub-MS_CLIP-0F62FE?logo=github)](https://github.com/IBM/MS-CLIP)
+
 # Llama3-MS-CLIP ViT-B/16
 
-More information coming soon. Stay tuned!
+Llama3-MS-CLIP is the first Vision-Language Model in the CLIP family that understands multispectral imagery.
+It is trained on one million image-text pairs from the SSL4EO-S12-v1.1 dataset with generated captions.
+
+## Architecture
+
+The CLIP model consists of two encoders for text and images. We extended the RGB patch embeddings to multispectral input and initialized the weights of the additional input channels with zeros. During the continual pre-training, the images and texts of each batch are encoded and combined. The loss increases the similarity of matching pairs while decreasing other combinations.
+
+![Architecture](assets/clip_architecture.png)
+
+## Evaluation
+
+We evaluated Llama3-MS-CLIP with zero-shot classification and text-to-image retrieval results, measured in accuracy (%) ↑ and mAP@100 (%) ↑, respectively. 
+The following Figure compares our model with the OpenCLIP baselines and other EO-VLMs. 
+We applied a smoothed min-max scaling and annotated the lowest and highest scores. 
+Our multispectral CLIP model is outperforming other RGB-based models on most benchmarks.
+
+![Benchmarking](assets/benchmarking.png)
+
+
+## Usage
+
+You can use the model out of the box for zero-shot classification and text-to-image retrieval with Sentinel-2 L2A images.
+
+### Setup 
+
+Clone the repository and set up a new env:
+```shell
+git clone https://github.com/IBM/MS-CLIP
+cd MS-CLIP
+python -m venv venv
+source venv/bin/activate
+pip install -e .
+```
+
+Run zero-shot classification with:
+```shell
+python inference.py --run-classification \
+    --model-name "Llama3-MS-CLIP-Base" \
+    --images "path/to/sentinel2_files.tif" \
+    --class-names "class 1" "class 2" "class 3"
+```
+
+Run text-to-image retrieval with:
+```shell
+python inference.py --run_retrieval\
+    --model-name "Llama3-MS-CLIP-Base" \
+    --images "path/to/sentinel2_files.tif" \
+    --query "Your query text" \
+    --top-k 5  # Number of retrieved images
+```
+
+More information is provided in the GitHub repository [MS-CLIP](https://github.com/IBM/MS-CLIP).
+
+## Citation
+
+Please cite the following paper, if you use Llama3-MS-CLIP in your research:
+
+```
+@article{marimo2025beyond,
+  title={Beyond the Visible: Multispectral Vision-Language Learning for Earth Observation},
+  author={Marimo, Clive Tinashe and Blumenstiel, Benedikt and Nitsche, Maximilian and Jakubik, Johannes and Brunschwiler, Thomas},
+  journal={arXiv preprint arXiv:2503.15969},
+  year={2025}
+}
+```
+
+## License
+
+Built with Meta Llama 3. 
+
+While the model itself is not based on Llama 3 but OpenCLIP B/16, it is trained on captions generated by a Llama 3-derivative model (License: https://github.com/meta-llama/llama3/blob/main/LICENSE).