codealchemist01's picture
Upload models/__init__.py with huggingface_hub
7049b0e verified
raw
history blame contribute delete
315 Bytes
"""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'
]