Update README.md

README.md

@@ -20,7 +20,6 @@ pip install samv2 huggingface_hub
 Each model requires different classes to infer. 
 
 ```python
-
 from huggingface_hub import hf_hub_download
 from sam2.build_sam import build_sam2
 from sam2.sam2_image_predictor import SAM2ImagePredictor
@@ -41,6 +40,24 @@ masks = predictor.predict(box=box,
 multimask_output=False)
 ```
 
+For automatic mask generation:
+
+```python
+from huggingface_hub import hf_hub_download
+from sam2.build_sam import build_sam2
+from sam2.automatic_mask_generator import SAM2AutomaticMaskGenerator
+
+hf_hub_download(repo_id = "merve/sam2-hiera-tiny", filename="sam2_hiera_tiny.pt", local_dir = "./")
+sam2_checkpoint = "../checkpoints/sam2_hiera_tiny.pt"
+model_cfg = "sam2_hiera_t.yaml"
+
+sam2 = build_sam2(model_cfg, sam2_checkpoint, device ='cuda', apply_postprocessing=False)
+
+mask_generator = SAM2AutomaticMaskGenerator(sam2)
+masks = mask_generator.generate(image)
+```
+
+
 ## Resources
 
 The team behind SAM2 made example notebooks for all tasks.