models/__init__.py

from .visual_model import VisualPersonalityModel, build_backbone
from .audio_model import (
    FrozenExtractor,
    LinearProjection,
    AGRU,
    AttentionPooling,
    PersonalityRegressor,
)
from .fusion_model import FusionMLP

__all__ = [
    "VisualPersonalityModel",
    "build_backbone",
    "FrozenExtractor",
    "LinearProjection",
    "AGRU",
    "AttentionPooling",
    "PersonalityRegressor",
    "FusionMLP",
]