Update config.py

config.py

@@ -1,6 +1,10 @@
 from pydantic import BaseSettings, Field
 import logging
 
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
 class Settings(BaseSettings):
     # Default values for the processor and model
     processor: str = Field(default="senga-ml/donut-v16")
@@ -30,14 +34,14 @@ class Settings(BaseSettings):
         # You had "donut-16" in some places and "donut-v16" in others
         shipper_model_map = {
             # Group 1 models - using donut-v16
-            61: {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"},
-            81: {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"},
-            139: {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"},
+            "61": {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"},
+            "81": {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"},
+            "139": {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"},
             
             # Group 2 models - using donut-v17
-            165: {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"},
-            127: {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"},
-            145: {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"},
+            "165": {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"},
+            "127": {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"},
+            "145": {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"},
         }
         
         previous_model = self.model
@@ -58,7 +62,7 @@ class Settings(BaseSettings):
         
         return self.model, self.processor
 
-# Create a single instance
+# Create a singleton instance
 settings = Settings()
 logger.info(f"Initial model setup: {settings.model}")