settings.py 445 Bytes
from pathlib import Path
from pydantic import BaseModel


class ServiceSettings(BaseModel):
    data_dir: str = "data/synthetic_v2"
    model_path: str = "data/models_v3/best_model.pt"
    index_prefix: str = "data/index_v3/reference"
    device: str = "cpu"

    def ensure_parent_dirs(self):
        for p in [Path(self.data_dir), Path(self.model_path).parent, Path(self.index_prefix).parent]:
            p.mkdir(parents=True, exist_ok=True)