Spaces:
Sleeping
Sleeping
Update config.py
Browse files
config.py
CHANGED
@@ -12,6 +12,13 @@ class Settings(BaseSettings):
|
|
12 |
127: "donut-v17",
|
13 |
}
|
14 |
space_base: str = "senga-ml"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
15 |
|
16 |
def get_model_url(self, shipper_id: int) -> str:
|
17 |
model = self.model_map.get(shipper_id, "default-model")
|
|
|
12 |
127: "donut-v17",
|
13 |
}
|
14 |
space_base: str = "senga-ml"
|
15 |
+
dataset: str = "senga-ml/dnotes-data-v6"
|
16 |
+
base_config: str = "naver-clova-ix/donut-base"
|
17 |
+
base_processor: str = "naver-clova-ix/donut-base"
|
18 |
+
base_model: str = "naver-clova-ix/donut-base"
|
19 |
+
inference_stats_file: str = "data/donut_inference_stats.json"
|
20 |
+
training_stats_file: str = "data/donut_training_stats.json"
|
21 |
+
evaluate_stats_file: str = "data/donut_evaluate_stats.json"
|
22 |
|
23 |
def get_model_url(self, shipper_id: int) -> str:
|
24 |
model = self.model_map.get(shipper_id, "default-model")
|