Spaces:
Running
Running
Update config.py
Browse files
config.py
CHANGED
@@ -2,7 +2,6 @@ from pydantic import BaseSettings
|
|
2 |
from typing import Dict
|
3 |
|
4 |
class Settings(BaseSettings):
|
5 |
-
# A dictionary to map shipper_id to the appropriate model
|
6 |
model_map: Dict[int, str] = {
|
7 |
61: "donut-v16",
|
8 |
81: "donut-v16",
|
|
|
2 |
from typing import Dict
|
3 |
|
4 |
class Settings(BaseSettings):
|
|
|
5 |
model_map: Dict[int, str] = {
|
6 |
61: "donut-v16",
|
7 |
81: "donut-v16",
|