config.py

from pydantic_settings import BaseSettings


class Settings(BaseSettings):
    app_name: str = "CGAN Image Colourization API"
    version: str = "2024.04.15"
    image_size: int = 320
    device: str = "cpu"


settings = Settings()