model_cache/__init__.py

from .custom_pipeline import IndianLawRAGPipeline

def transformers_module():
    '''Register the custom pipeline'''
    from transformers.pipelines import PIPELINE_REGISTRY
    PIPELINE_REGISTRY.register_pipeline(
        "indian-law-rag",
        pipeline_class=IndianLawRAGPipeline,
        pt_model=None,  # Works with any PT model
    )