initial commit

app.py

@@ -12,7 +12,15 @@ os.chdir("segment-anything-2")
 # Install the package
 subprocess.run(shlex.split("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)
 
 def run(image):
     return None
@@ -31,15 +39,3 @@ demo = gr.Interface(
     allow_duplication=True,
 )
 demo.queue().launch()
-
-
-
-# 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)