v2run commited on
Commit
2784359
1 Parent(s): 618948c

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +19 -0
config.py CHANGED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from pydantic import BaseSettings
2
+ import os
3
+
4
+
5
+ class Settings(BaseSettings):
6
+ huggingface_key: str = os.environ.get("huggingface_key")
7
+ sparrow_key: str = os.environ.get("sparrow_key")
8
+ processor: str = "katanaml-org/invoices-donut-model-v1"
9
+ model: str = "katanaml-org/invoices-donut-model-v1"
10
+ dataset: str = "katanaml-org/invoices-donut-data-v1"
11
+ base_config: str = "naver-clova-ix/donut-base"
12
+ base_processor: str = "naver-clova-ix/donut-base"
13
+ base_model: str = "naver-clova-ix/donut-base"
14
+ inference_stats_file: str = "data/donut_inference_stats.json"
15
+ training_stats_file: str = "data/donut_training_stats.json"
16
+ evaluate_stats_file: str = "data/donut_evaluate_stats.json"
17
+
18
+
19
+ settings = Settings()