serenarolloh commited on
Commit
6b214e4
·
verified ·
1 Parent(s): 8cf136d

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +7 -0
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")