app.py

import subprocess
import shlex
import os

# Clone the repository
subprocess.run(shlex.split("git clone https://github.com/facebookresearch/segment-anything-2.git"))

# Change to the repository directory
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)