src/__init__.py

# src/__init__.py

# Biar bisa langsung import fungsi/kelas dari dataset dan model
from .dataset import CustomDataset
from .model import get_vit_model

__all__ = ["CustomDataset", "get_vit_model"]