Update models.py

models.py

@@ -45,21 +45,40 @@ class BagelAPIAnalyzer(BaseImageAnalyzer):
         self.client = None
         self.space_url = "Malaji71/Bagel-7B-Demo"
         self.api_endpoint = "/image_understanding"
+        self.hf_token = os.getenv("HF_TOKEN")  # Get token from environment/secrets
         
     def initialize(self) -> bool:
-        """Initialize BAGEL API client"""
+        """Initialize BAGEL API client with authentication"""
         if self.is_initialized:
             return True
         
         try:
             logger.info("Initializing BAGEL API client...")
-            self.client = Client(self.space_url)
+            
+            # Initialize client with token if available (for private spaces)
+            if self.hf_token:
+                logger.info("Using HF token for private space access")
+                self.client = Client(self.space_url, hf_token=self.hf_token)
+            else:
+                logger.info("No HF token found, accessing public space")
+                self.client = Client(self.space_url)
+                
             self.is_initialized = True
             logger.info("BAGEL API client initialized successfully")
             return True
             
         except Exception as e:
             logger.error(f"BAGEL API client initialization failed: {e}")
+            # If private space fails, try without token as fallback
+            if self.hf_token:
+                logger.info("Retrying without token...")
+                try:
+                    self.client = Client(self.space_url)
+                    self.is_initialized = True
+                    logger.info("BAGEL API client initialized successfully (fallback to public)")
+                    return True
+                except Exception as e2:
+                    logger.error(f"Fallback initialization also failed: {e2}")
             return False
 
     def _extract_camera_setup(self, description: str) -> Optional[str]: