src/models/__init__.py

"""
============================================================================
MODELS MODULE - Gestión de Modelos de Transformers
============================================================================

Este módulo proporciona utilidades para cargar y usar modelos de HuggingFace,
específicamente DistilBERT para análisis de sentimientos.

Componentes principales:
    - ModelLoader: Clase para cargar y predecir con DistilBERT

Uso:
    from src.models import ModelLoader
    from src.config import Config
    
    config = Config()
    model = ModelLoader(config)
    result = model.predict("This movie is great!")

Autor: Proyecto Módulo II
Fecha: 2025-01-15
============================================================================
"""

# Importar clase principal
from .model_loader import ModelLoader

# Definir qué se exporta cuando se hace "from src.models import *"
__all__ = [
    'ModelLoader'
]

# Información del módulo
__version__ = '1.0.0'
__author__ = 'Estudiante Módulo II'

# Logging
import logging
logger = logging.getLogger(__name__)
logger.info("✅ Módulo 'models' cargado correctamente")