import os


class Config:
    UPLOAD_FOLDER = 'static/uploads'
    RESULT_FOLDER = 'static/results'
    HARDCODED_IMAGES = 'training/memory'
    ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg'}
    
    # Model configuration
    MODEL_PATH = 'models/memory_detector.pt'
    CONFIDENCE_THRESHOLD = 0.5