File size: 142 Bytes
0fe8b25
aa16163
0fe8b25
 
1
2
3
4
from custom_image_to_text_pipeline import ImageToTextPipeline

def get_inference_handler(model_dir):
    return ImageToTextPipeline(model_dir)