Update config.py

config.py

@@ -1,18 +1,9 @@
 from pydantic import BaseSettings
-from typing import Dict
 
 class Settings(BaseSettings):
-    model_map: Dict[int, str] = {
-        61: "donut-v16",
-        81: "donut-v16",
-        139: "donut-v16",
-        165: "donut-v17",
-        145: "donut-v17",
-        127: "donut-v17",
-    }
-    space_base: str = "senga-ml"
-    processor: str = "senga-ml/donut-v16"
-    model: str = "senga-ml/donut-v16"
+    # Default values for the processor and model
+    processor: str = "senga-ml/naivas_lpos"
+    model: str = "senga-ml/naivas_lpos"
     dataset: str = "senga-ml/dnotes-data-v6"
     base_config: str = "naver-clova-ix/donut-base"
     base_processor: str = "naver-clova-ix/donut-base"
@@ -20,9 +11,40 @@ class Settings(BaseSettings):
     inference_stats_file: str = "data/donut_inference_stats.json"
     training_stats_file: str = "data/donut_training_stats.json"
     evaluate_stats_file: str = "data/donut_evaluate_stats.json"
+    
+    # The shipper_id to dynamically select model and processor
+    shipper_id: str = "default_shipper"
+
+    # Function to dynamically select model and processor based on shipper_id
+    def set_model(self):
+        if self.shipper_id == "61":
+            self.model = "senga-ml/donut-16"
+            self.processor = "senga-ml/donut-16"
+        elif self.shipper_id == "81":
+            self.model = "senga-ml/donut-16"
+            self.processor = "senga-ml/donut-16"
+        elif self.shipper_id == "139":
+            self.model = "senga-ml/donut-16"
+            self.processor = "senga-ml/donut-16"
+        elif self.shipper_id == "165"   
+            self.model = "senga-ml/donut-17"
+            self.processor = "senga-ml/donut-17"
+        elif self.shipper_id == "127"
+            self.model = "senga-ml/donut-17"
+            self.processor = "senga-ml/donut-17"
+        elif self.shipper_id == "145"
+            self.model = "senga-ml/donut-17"
+            self.processor = "senga-ml/donut-17"
+        else:
+            self.model = self.base_model  # Default to base model
+            self.processor = self.base_processor  # Default to base processor
+
+    # Initialize dynamic model selection when settings are loaded
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+        self.set_model()  # Set the model based on shipper_id
 
-    def get_model_url(self, shipper_id: int) -> str:
-        model = self.model_map.get(shipper_id, "senga-ml/donut-v16")  
-        return f"https://huggingface.co/{self.space_base}/{model}"
 
 settings = Settings()
+
+