from custom_image_to_text_pipeline import ImageToTextPipeline

def get_inference_handler(model_dir):
    return ImageToTextPipeline(model_dir)