hydra dependency removed

app.py

@@ -1,5 +1,4 @@
 import torch
-from sam2.build_sam import build_sam2
 from sam2.automatic_mask_generator import SAM2AutomaticMaskGenerator
 from PIL import Image
 from matplotlib import pyplot as plt
@@ -66,13 +65,10 @@ def get_response(image):
     return overlay_masks_on_image(image,masks)
 
 def download_checkpoint():
-    os.system('wget https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_large.pt')
+    os.system('gdown 1RHSO8lHko3IK3dmABOzFDJuq7wmKVcun')
 
 
 
-def load_model_cfg():
-    return SAM2AutomaticMaskGenerator(build_sam2(model_cfg, checkpoint, device='cpu'))
-
 
 if __name__ == "__main__":
 
@@ -85,15 +81,13 @@ if __name__ == "__main__":
         title="Segmenting Microscopic images with Segment Anything",
         description="Segmenting Microscopic images with Meta Segment Anything")
 
-
-    checkpoint = "sam2.1_hiera_large.pt"
-    model_cfg =  "sam2.1_hiera_l.yaml"
-
-    if not os.path.exists(checkpoint):
-        print('Downloading  checkpoint')
+    model_path='model.pth'
+    if not os.path.exists(model_path):
+        print('Downloading  model with weights')
         download_checkpoint()
-        print('Checkpoint Downloaded')
+        print('Model with weights Downloaded')
 
-    mask_generator = load_model_cfg()
+    model = torch.load(model_path, map_location="cpu", weights_only=False)
+    mask_generator = SAM2AutomaticMaskGenerator(model)
 
     iface.launch()

requirements.txt

@@ -1,2 +1,4 @@
-#hydra-core==1.3.2
-antlr4-python3-runtime==4.9.3
+torch
+torchvision
+opencv_python
+gdown