sparrow-API / config.py
joemaatouk's picture
Update config.py
557befb
from pydantic import BaseSettings
import os
class Settings(BaseSettings):
huggingface_key: str = "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"
sparrow_key: str = "TestKey"
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()