38 lines
832 B
Python
38 lines
832 B
Python
from pydantic_settings import BaseSettings, SettingsConfigDict
|
|
|
|
class Settings(BaseSettings):
|
|
APP_NAME: str = "AI Moderation"
|
|
|
|
APP_VERSION: str = "1.0.0"
|
|
|
|
HOST: str = "0.0.0.0"
|
|
|
|
PORT: int = 8000
|
|
|
|
DEBUG: bool = True
|
|
|
|
TEXT_TOXIC_THRESHOLD: float = 0.90
|
|
|
|
# ===== AI =====
|
|
|
|
TEXT_MODEL: str = "textdetox/bert-multilingual-toxicity-classifier"
|
|
|
|
IMAGE_MODEL: str = "Falconsai/nsfw_image_detection"
|
|
NSFW_THRESHOLD: float = 0.85
|
|
IMAGE_CLIP_MODEL: str = "openai/clip-vit-base-patch32"
|
|
CLIP_THRESHOLD: float = 0.75
|
|
WEAPON_MODEL: str = "google/owlv2-base-patch16-ensemble"
|
|
WEAPON_THRESHOLD: float = 0.40
|
|
WEAPON_MIN_AREA_FRACTION: float = 0.005
|
|
MODEL_CACHE_DIR: str = "./models"
|
|
|
|
DEVICE: str = "auto"
|
|
|
|
model_config = SettingsConfigDict(
|
|
env_file=".env",
|
|
extra="ignore"
|
|
)
|
|
|
|
|
|
settings = Settings()
|