serenarolloh commited on
Commit
461c1a0
·
verified ·
1 Parent(s): c84055e

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +6 -6
config.py CHANGED
@@ -26,12 +26,12 @@ class Settings(BaseSettings):
26
  # Function to dynamically select model and processor based on shipper_id
27
  def set_model(self):
28
  shipper_model_map = {
29
- "61": {"model": "senga-ml/donut-16", "processor": "senga-ml/donut-16"},
30
- "81": {"model": "senga-ml/donut-16", "processor": "senga-ml/donut-16"},
31
- "139": {"model": "senga-ml/donut-16", "processor": "senga-ml/donut-16"},
32
- "165": {"model": "senga-ml/donut-17", "processor": "senga-ml/donut-17"},
33
- "127": {"model": "senga-ml/donut-17", "processor": "senga-ml/donut-17"},
34
- "145": {"model": "senga-ml/donut-17", "processor": "senga-ml/donut-17"},
35
  }
36
 
37
  config = shipper_model_map.get(
 
26
  # Function to dynamically select model and processor based on shipper_id
27
  def set_model(self):
28
  shipper_model_map = {
29
+ "61": {"model": "senga-ml/donut-16", "processor": "senga-ml/donut-v16"},
30
+ "81": {"model": "senga-ml/donut-16", "processor": "senga-ml/donut-v16"},
31
+ "139": {"model": "senga-ml/donut-16", "processor": "senga-ml/donut-v16"},
32
+ "165": {"model": "senga-ml/donut-17", "processor": "senga-ml/donut-v17"},
33
+ "127": {"model": "senga-ml/donut-17", "processor": "senga-ml/donut-v17"},
34
+ "145": {"model": "senga-ml/donut-17", "processor": "senga-ml/donut-v17"},
35
  }
36
 
37
  config = shipper_model_map.get(