config.py

from pydantic import BaseSettings


class Settings(BaseSettings):
    huggingface_key: str = "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"
    processor: str = "senga-ml/donut-training-v4"
    model: str = "senga-ml/donut-training-v4"
    dataset: str = "senga-ml/dnotes-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()