Update config.py

config.py

@@ -1,8 +1,8 @@
 from pydantic import BaseSettings
 from typing import Dict
 
-
 class Settings(BaseSettings):
+    # A dictionary to map shipper_id to the appropriate model
     model_map: Dict[int, str] = {
         61: "donut-v16",
         81: "donut-v16",
@@ -14,8 +14,7 @@ class Settings(BaseSettings):
     space_base: str = "senga-ml"
 
     def get_model_url(self, shipper_id: int) -> str:
-        model = self.model_map.get(shipper_id, "default-model")
+        model = self.model_map.get(shipper_id, "default-model")  
         return f"https://{self.space_base}/{model}"
 
-
 settings = Settings()