Update README.md

README.md

@@ -19,6 +19,8 @@ pip install samv2 huggingface_hub
 
 Each model requires different classes to infer. 
 
+For prompting:
+
 ```python
 
 from huggingface_hub import hf_hub_download
@@ -41,6 +43,23 @@ 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-small", filename="sam2_hiera_small.pt", local_dir = "./")
+ckpt = f"./sam2_hiera_small.pt"
+config = "sam2_hiera_s.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.