__init__.py

"""AshishOCR model package - Custom OCR model based on vision-language architecture."""

from transformers import AutoConfig, AutoModel, AutoModelForCausalLM, AutoProcessor, AutoImageProcessor

from .configuration_ashish_ocr import (
    AshishOcrConfig,
    AshishOcrTextConfig,
    AshishOcrVisionConfig,
)
from .modeling_ashish_ocr import (
    AshishOcrPreTrainedModel,
    AshishOcrTextModel,
    AshishOcrVisionEncoder,
    AshishOcrForConditionalGeneration,
)
from .processing_ashish_ocr import (
    AshishOcrImageProcessor,
    AshishOcrProcessor,
)

# Register model with transformers Auto classes
AutoConfig.register("ashish_ocr", AshishOcrConfig)
AutoConfig.register("ashish_ocr_text", AshishOcrTextConfig)
AutoConfig.register("ashish_ocr_vision", AshishOcrVisionConfig)

AutoModel.register(AshishOcrConfig, AshishOcrForConditionalGeneration)
AutoModelForCausalLM.register(AshishOcrConfig, AshishOcrForConditionalGeneration)

AutoImageProcessor.register(AshishOcrConfig, AshishOcrImageProcessor)
AutoProcessor.register(AshishOcrConfig, AshishOcrProcessor)

__all__ = [
    "AshishOcrConfig",
    "AshishOcrTextConfig",
    "AshishOcrVisionConfig",
    "AshishOcrPreTrainedModel",
    "AshishOcrTextModel",
    "AshishOcrVisionEncoder",
    "AshishOcrForConditionalGeneration",
    "AshishOcrImageProcessor",
    "AshishOcrProcessor",
]