File size: 940 Bytes
60c56d7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
"""
Configuration settings for the application
"""
import os
from pydantic_settings import BaseSettings
class Settings(BaseSettings):
"""Application settings"""
# Firebase settings
ENABLE_APP_CHECK: bool = os.getenv("ENABLE_APP_CHECK", "true").lower() == "true"
FIREBASE_CREDENTIALS_PATH: str = os.getenv(
"FIREBASE_CREDENTIALS_PATH",
"colorize-662df-firebase-adminsdk-fbsvc-e080668793.json"
)
# API settings
BASE_URL: str = os.getenv("BASE_URL", "http://localhost:8000")
# Model settings
MODEL_ID: str = os.getenv("MODEL_ID", "rsortino/ColorizeNet")
NUM_INFERENCE_STEPS: int = int(os.getenv("NUM_INFERENCE_STEPS", "20"))
# Storage settings
UPLOAD_DIR: str = os.getenv("UPLOAD_DIR", "uploads")
RESULT_DIR: str = os.getenv("RESULT_DIR", "results")
class Config:
env_file = ".env"
case_sensitive = False
settings = Settings()
|