Update README.md

README.md

@@ -3,7 +3,7 @@ license: apache-2.0
 pipeline_tag: mask-generation
 library_name: sam2
 ---
-Model cloned from this [repo](https://github.com/facebookresearch/segment-anything-2/)
+Model cloned from this [repo](https://github.com/facebookresearch/segment-anything-2/) and will be finetuned.
 Repository for SAM 2: Segment Anything in Images and Videos, a foundation model towards solving promptable visual segmentation in images and videos from FAIR. See the [SAM 2 paper](https://arxiv.org/abs/2408.00714) for more information.
 
 The official code is publicly release in this [repo](https://github.com/facebookresearch/segment-anything-2/).
@@ -16,7 +16,7 @@ For image prediction:
 import torch
 from sam2.sam2_image_predictor import SAM2ImagePredictor
 
-predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2-hiera-tiny")
+predictor = SAM2ImagePredictor.from_pretrained("iloncka/culico-net-segm-v1-nano")
 
 with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
     predictor.set_image(<your_image>)
@@ -29,7 +29,7 @@ For video prediction:
 import torch
 from sam2.sam2_video_predictor import SAM2VideoPredictor
 
-predictor = SAM2VideoPredictor.from_pretrained("facebook/sam2-hiera-tiny")
+predictor = SAM2VideoPredictor.from_pretrained("iloncka/culico-net-segm-v1-nano")
 
 with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
     state = predictor.init_state(<your_video>)