Update config.py

config.py

@@ -2,8 +2,20 @@ from pydantic import BaseSettings
 
 
 class Settings(BaseSettings):
-    processor: str = "senga-ml/naivas_grns"
-    model: str = "senga-ml/naivas_grns"
+     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"
+
+    def get_model_url(self, shipper_id: int) -> str:
+        model = self.model_map.get(shipper_id, "default-model")
+        return f"https://{self.space_base}/{model}"
+        
     dataset: str = "senga-ml/dnotes-data-v6"
     base_config: str = "naver-clova-ix/donut-base"
     base_processor: str = "naver-clova-ix/donut-base"