update the secret

sorghum_pipeline/pipeline.py

@@ -33,6 +33,7 @@ class SorghumPipeline:
         self.vegetation_extractor = VegetationIndexExtractor()
         self.morphology_extractor = MorphologyExtractor()
         self.segmentation_manager = SegmentationManager(
+            model_name="briaai/RMBG-2.0",
             device=self.config.get_device(),
             trust_remote_code=True
         )

sorghum_pipeline/segmentation/manager.py

@@ -25,12 +25,17 @@ class SegmentationManager:
         self.threshold = threshold
         self.device = "cuda" if device == "auto" and torch.cuda.is_available() else device
         
+        # Get HF token from environment (set as Space secret)
+        import os
+        hf_token = os.environ.get("HF_TOKEN")
+        
         logger.info(f"Loading BRIA model: {model_name}")
         self.model = AutoModelForImageSegmentation.from_pretrained(
             model_name,
             trust_remote_code=trust_remote_code,
             cache_dir=cache_dir if cache_dir else None,
             local_files_only=local_files_only,
+            token=hf_token,
         ).eval().to(self.device)
         
         self.transform = transforms.Compose([