Spaces:
Sleeping
Sleeping
Update config.py
Browse files
config.py
CHANGED
@@ -30,14 +30,28 @@ class Settings(BaseSettings):
|
|
30 |
def set_model(self):
|
31 |
shipper_model_map = {
|
32 |
# Shippers using donut-v16
|
33 |
-
"61": {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"},
|
34 |
-
"81": {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"},
|
35 |
-
"139": {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"},
|
|
|
|
|
|
|
|
|
36 |
|
37 |
# Shippers using donut-v17
|
38 |
-
"165": {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"},
|
39 |
-
"127": {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"},
|
40 |
-
"145": {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
41 |
}
|
42 |
|
43 |
previous_model = self.model
|
|
|
30 |
def set_model(self):
|
31 |
shipper_model_map = {
|
32 |
# Shippers using donut-v16
|
33 |
+
"61": {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"}, # Tropical Heat
|
34 |
+
"81": {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"}, # Clique
|
35 |
+
"139": {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"}, # Winnie's
|
36 |
+
"67": {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"}, # Skypex
|
37 |
+
"193": {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"}, # Storesome
|
38 |
+
"195": {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"}, # On Demand
|
39 |
+
"194": {"model": "senga-ml/donut-v16", "processor": "senga-ml/donut-v16"}, # Healthy U
|
40 |
|
41 |
# Shippers using donut-v17
|
42 |
+
"165": {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"}, # Loki
|
43 |
+
"127": {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"}, # Jumbo Spices
|
44 |
+
"145": {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"}, # Carolines
|
45 |
+
"80": {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"}, # Portcross
|
46 |
+
"220": {"model": "senga-ml/donut-v17", "processor": "senga-ml/donut-v17"}, # Zulu
|
47 |
+
|
48 |
+
# Shippers using donut-v18
|
49 |
+
"177": {"model": "senga-ml/donut-v18", "processor": "senga-ml/donut-v18"}, # Smart Global
|
50 |
+
"216": {"model": "senga-ml/donut-v18", "processor": "senga-ml/donut-v18"}, # Melvins
|
51 |
+
"182": {"model": "senga-ml/donut-v18", "processor": "senga-ml/donut-v18"}, # Roshni
|
52 |
+
"226": {"model": "senga-ml/donut-v18", "processor": "senga-ml/donut-v18"}, # Fayaz
|
53 |
+
"145": {"model": "senga-ml/donut-v18", "processor": "senga-ml/donut-v18"}, # Beyond Opinion
|
54 |
+
"232": {"model": "senga-ml/donut-v18", "processor": "senga-ml/donut-v18"}, # Gel x
|
55 |
}
|
56 |
|
57 |
previous_model = self.model
|