HtmlTocode / handler.py
marutitecblic's picture
Update handler.py
0fe8b25 verified
from custom_image_to_text_pipeline import ImageToTextPipeline
def get_inference_handler(model_dir):
return ImageToTextPipeline(model_dir)