initial commit

app.py

@@ -1,18 +1,22 @@
-import shlex
 import subprocess
+import shlex
+import os
 
-subprocess.run(
-    shlex.split(
-        "git clone https://github.com/facebookresearch/segment-anything-2.git && cd segment-anything-2 && pip install -e ."
-    )
-)
+# Clone the repository
+subprocess.run(shlex.split("git clone https://github.com/facebookresearch/segment-anything-2.git"))
 
-import torch
-from sam2.build_sam import build_sam2
-from sam2.sam2_image_predictor import SAM2ImagePredictor
+# Change to the repository directory
+os.chdir("segment-anything-2")
 
-DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-CHECKPOINT = f"checkpoints/sam2_hiera_large.pt"
-CONFIG = "sam2_hiera_l.yaml"
+# Install the package
+subprocess.run(shlex.split("pip install -e ."))
 
-sam2_model = build_sam2(CONFIG, CHECKPOINT, device=DEVICE)
+# 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)