src/config.py

import os
from dotenv import load_dotenv

load_dotenv()

class Config:
    """Application configuration class."""

    # Hugging Face Configuration
    HF_TOKEN = os.getenv("HF_TOKEN")

    # Model Configuration
    DETR_MODEL_NAME = "facebook/detr-resnet-50"
    QWEN_MODEL_NAME = "Qwen/Qwen2.5-Omni-7B"

    @classmethod
    def validate_config(cls):
        """Validate the configuration values."""
        if not cls.HF_TOKEN:
            raise ValueError("HF_TOKEN is not set in the environment variables.")
        
# Validate configuration on import
Config.validate_config()