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)