Update README.md

README.md

@@ -1,5 +1,10 @@
+---
+license: apache-2.0
+---
 # Segment Anything
 
+This repository is the mirror of the official [Segment Anything repository](https://github.com/facebookresearch/segment-anything). 
+
 **[Meta AI Research, FAIR](https://ai.facebook.com/research/)**
 
 [Alexander Kirillov](https://alexander-kirillov.github.io/), [Eric Mintun](https://ericmintun.github.io/), [Nikhila Ravi](https://nikhilaravi.com/), [Hanzi Mao](https://hanzimao.me/), Chloe Rolland, Laura Gustafson, [Tete Xiao](https://tetexiao.com), [Spencer Whitehead](https://www.spencerwhitehead.com/), Alex Berg, Wan-Yen Lo, [Piotr Dollar](https://pdollar.github.io/), [Ross Girshick](https://www.rossgirshick.info/)
@@ -15,6 +20,20 @@ The **Segment Anything Model (SAM)** produces high quality object masks from inp
   <img src="https://raw.githubusercontent.com/facebookresearch/segment-anything/main/assets/masks2.jpg" width="61.5%" /> 
 </p>
 
+## Download the checkpoints
+
+```bash
+pip install huggingface_hub
+```
+
+Then,
+
+```python
+from huggingface_hub import hf_hub_download
+
+chkpt_path = hf_hub_download("facebook/segment-anything", "checkpoints/sam_vit_b_01ec64.pth")
+```
+
 ## Installation
 
 The code requires `python>=3.8`, as well as `pytorch>=1.7` and `torchvision>=0.8`. Please follow the instructions [here](https://pytorch.org/get-started/locally/) to install both PyTorch and TorchVision dependencies. Installing both PyTorch and TorchVision with CUDA support is strongly recommended.