File size: 735 Bytes
ce1dd07
 
 
 
 
 
 
 
 
dfcd6b0
 
 
 
ce1dd07
 
dfcd6b0
ce1dd07
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
from pydantic import BaseSettings
import os


class Settings(BaseSettings):
    huggingface_key: str = os.environ.get("huggingface_key")
    sparrow_key: str = os.environ.get("sparrow_key")
    processor: str = "katanaml-org/invoices-donut-model-v1"
    model: str = "katanaml-org/invoices-donut-model-v1"
    dataset: str = "katanaml-org/invoices-donut-data-v1"
    base_config: str = "naver-clova-ix/donut-base"
    base_processor: str = "naver-clova-ix/donut-base"
    base_model: str = "naver-clova-ix/donut-base"
    inference_stats_file: str = "data/donut_inference_stats.json"
    training_stats_file: str = "data/donut_training_stats.json"
    evaluate_stats_file: str = "data/donut_evaluate_stats.json"


settings = Settings()