add sam2

app.py

@@ -6,6 +6,7 @@ from torchvision import transforms
 from transformers import AutoModelForImageSegmentation
 from diffusers import FluxFillPipeline
 from PIL import Image, ImageOps
+from sam2.sam2_image_predictor import SAM2ImagePredictor
 
 torch.set_float32_matmul_precision(["high", "highest"][0])
 
@@ -14,6 +15,8 @@ birefnet = AutoModelForImageSegmentation.from_pretrained(
 )
 birefnet.to("cuda")
 
+predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2-hiera-tiny", device="cpu")
+
 transform_image = transforms.Compose(
     [
         transforms.Resize((1024, 1024)),

requirements.txt

@@ -19,3 +19,4 @@ kornia
 huggingface_hub
 sentencepiece
 einops
+git+https://github.com/facebookresearch/sam2.git