Update modules/masking_module.py

modules/masking_module.py

@@ -34,13 +34,14 @@ def load_sam_image_model(
     model = build_sam2(config, checkpoint, device=device)
     return SAM2ImagePredictor(sam_model=model)
 
-SAM_IMAGE_MODEL = load_sam_image_model(device=device)
+
 @spaces.GPU
 def run_sam_inference(
     model: Any,
     image: Image,
     detections: sv.Detections
 ) -> sv.Detections:
+    
     image = np.array(image.convert("RGB"))
     model.set_image(image)
     print(type(detections.xyxy),detections.xyxy)
@@ -151,7 +152,7 @@ def masking_process(image,obj):
     # obj = "Tiger"
 
     Florence_results = florence2(image,task_prompt, text_input=obj)
-
+    SAM_IMAGE_MODEL = load_sam_image_model(device=device)
     detections = sv.Detections.from_lmm(
                 lmm=sv.LMM.FLORENCE_2,
                 result=Florence_results,