Spaces:
Sleeping
Sleeping
File size: 729 Bytes
c2d58b3 0669c63 6b214e4 e26f64e 1b3cfd8 ac89508 1b3cfd8 cadf158 ac89508 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
from pydantic import BaseSettings
class Settings(BaseSettings):
dataset: str "senga-ml/dnotes-dataset-v6"
base_config: str = "naver-clova-ix/donut-base"
base_processor: str = "naver-clova-ix/donut-base"
base_model: str = "naver-clova-ix/donut-base"
def get_model_config(self, shipper_id: str):
if shipper_id in ["61", "81", "139"]:
model = "senga-ml/donut-16"
processor = "senga-ml/donut-16"
elif shipper_id in ["165", "127", "145"]:
model = "senga-ml/donut-17"
processor = "senga-ml/donut-17"
else:
model = self.base_model
processor = self.base_processor
return model, processor
settings = Settings()
|