Update eval/grounded_sam/grounded_sam2_florence2_autolabel_pipeline.py

eval/grounded_sam/grounded_sam2_florence2_autolabel_pipeline.py

@@ -10,7 +10,7 @@ import sys
 
 from eval.grounded_sam.florence2.modeling_florence2 import Florence2ForConditionalGeneration
 from eval.grounded_sam.florence2.processing_florence2 import Florence2Processor
-from eval.grounded_sam.sam2.build_sam import build_sam2
+from eval.grounded_sam.sam2.build_sam import build_sam2, build_sam2_hf
 from eval.grounded_sam.sam2.sam2_image_predictor import SAM2ImagePredictor
 
 
@@ -61,7 +61,6 @@ class FlorenceSAM:
             
         FLORENCE2_MODEL_ID = os.getenv('FLORENCE2_MODEL_PATH', "microsoft/Florence-2-large")
         SAM2_CHECKPOINT = os.getenv('SAM2_MODEL_PATH', "facebook/sam2-hiera-large")
-        SAM2_CONFIG = "configs/sam2.1/sam2.1_hiera_l.yaml"
 
         self.florence2_model = Florence2ForConditionalGeneration.from_pretrained(
             FLORENCE2_MODEL_ID, 
@@ -70,7 +69,7 @@ class FlorenceSAM:
         self.florence2_processor = Florence2Processor.from_pretrained(
             FLORENCE2_MODEL_ID, 
         )
-        sam2_model = build_sam2(SAM2_CONFIG, SAM2_CHECKPOINT, device=self.device)
+        sam2_model = build_sam2_hf(SAM2_CHECKPOINT, device=self.device)
         self.sam2_predictor = SAM2ImagePredictor(sam2_model)
 
     def __str__(self):