imagedetector1 / TextGen /ConfigEnv.py
ShahanMalik's picture
Update TextGen/ConfigEnv.py
a9323b6 verified
raw
history blame contribute delete
531 Bytes
import os
class Config:
"""Configuration class for the AI Image Detection API"""
# Model configuration
MODEL_PATH = os.getenv('MODEL_PATH', 'src/best_model.keras')
MAX_FILE_SIZE = int(os.getenv('MAX_FILE_SIZE', 5 * 1024 * 1024)) # 5MB default
# API configuration
API_TITLE = os.getenv('API_TITLE', 'AI Image Detection API')
API_DESCRIPTION = os.getenv('API_DESCRIPTION', 'Detect AI-generated images using deep learning')
API_VERSION = os.getenv('API_VERSION', '1.0.0')
config = Config()