Update app.py

app.py

@@ -24,11 +24,9 @@ if not os.path.exists("weights/sam_vit_h_4b8939.pth"):
 # Add EfficientSAM to Python path
 sys.path.append(os.path.abspath("EfficientSAM-main"))
 
-subprocess.run(["pip", "install", "git+https://github.com/facebookresearch/segment-anything.git"])
-subprocess.run(["git", "clone", "https://github.com/yformer/EfficientSAM.git"])
-
+# Import SAM and EfficientSAM modules
 from segment_anything import sam_model_registry, SamAutomaticMaskGenerator
-from efficient_sam.build_efficient_sam_vits import build_efficient_sam_vits
+from efficient_sam.build_efficient_sam import build_efficient_sam_vits
 
 # Constants
 DEVICE = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
@@ -40,7 +38,7 @@ sam = sam_model_registry[MODEL_TYPE](checkpoint=CHECKPOINT_PATH).to(device=DEVIC
 mask_generator_sam = SamAutomaticMaskGenerator(sam)
 
 # Load EfficientSAM model
-with zipfile.ZipFile("EfficientSAM-main/weights/efficient_sam_vits.pt.zip", 'r') as zip_ref:
+with zipfile.ZipFile("weights/efficient_sam_vits.pt.zip", 'r') as zip_ref:
     zip_ref.extractall("weights")
 efficient_sam_vits_model = build_efficient_sam_vits()
 
@@ -167,6 +165,3 @@ interface = gr.Interface(
 )
 
 interface.launch(debug=True)
-
-
-