Fix import error

app.py

@@ -39,14 +39,22 @@ from efficient_sam.build_efficient_sam import build_efficient_sam_vits
 DEVICE = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
 MODEL_TYPE = "vit_h"
 CHECKPOINT_PATH = "weights/sam_vit_h_4b8939.pth"
+EFF_SAM_ZIP_PATH = "weights/efficient_sam_vits.pt.zip"
+EFF_SAM_EXTRACT_DIR = "weights/efficient_sam_vits"
 
 # Load SAM model
 sam = sam_model_registry[MODEL_TYPE](checkpoint=CHECKPOINT_PATH).to(device=DEVICE)
 mask_generator_sam = SamAutomaticMaskGenerator(sam)
 
-# Load EfficientSAM model
-with zipfile.ZipFile("weights/efficient_sam_vits.pt.zip", 'r') as zip_ref:
-    zip_ref.extractall("weights")
+# Download EfficientSAM weights if not present
+if not os.path.exists(EFF_SAM_ZIP_PATH):
+    download_file("https://example.com/path/to/efficient_sam_vits.pt.zip", EFF_SAM_ZIP_PATH)
+
+# Extract EfficientSAM weights if not already extracted
+if not os.path.exists(EFF_SAM_EXTRACT_DIR):
+    with zipfile.ZipFile(EFF_SAM_ZIP_PATH, 'r') as zip_ref:
+        zip_ref.extractall(EFF_SAM_EXTRACT_DIR)
+
 efficient_sam_vits_model = build_efficient_sam_vits()
 
 from segment_anything.utils.amg import (
@@ -174,3 +182,4 @@ interface = gr.Interface(
 interface.launch(debug=True)
 
 
+