models/__init__.py

"""Models package for Hybrid Food Classifier"""
from .cnn_branch import CNNBranch
from .vit_branch import ViTBranch
from .fusion_module import AdaptiveFusionModule
from .hybrid_model import HybridFoodClassifier

__all__ = [
    'HybridFoodClassifier',
    'CNNBranch', 
    'ViTBranch',
    'AdaptiveFusionModule'
]