initial commit

app.py

@@ -0,0 +1,18 @@
+import shlex
+import subprocess
+
+subprocess.run(
+    shlex.split(
+        "git clone https://github.com/facebookresearch/segment-anything-2.git && cd segment-anything-2 && pip install -e ."
+    )
+)
+
+import torch
+from sam2.build_sam import build_sam2
+from sam2.sam2_image_predictor import SAM2ImagePredictor
+
+DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+CHECKPOINT = f"checkpoints/sam2_hiera_large.pt"
+CONFIG = "sam2_hiera_l.yaml"
+
+sam2_model = build_sam2(CONFIG, CHECKPOINT, device=DEVICE)