config.py

import os

from pydantic import BaseModel


class Settings(BaseModel):
    hf_token: str = os.environ.get("hf_token")
    base_sd_model: str = os.environ.get("base_sd_model") or "stabilityai/stable-diffusion-xl-base-1.0"
    refiner_sd_model: str = os.environ.get("refiner_sd_model") or "stabilityai/stable-diffusion-xl-refiner-1.0"
    version: str = "0.1.0"
    url_version: str = "v1"
    prefix: str = "/v1/unik-ml"


settings = Settings()