model.py

from pix2tex.cli import LatexOCR
from pix2tex.model import get_model as pix2tex_model
from pix2tex.utils import Tokenizer

def get_model(weights='trained_model'):
    tokenizer = Tokenizer()
    model = pix2tex_model(checkpoint_path=weights)
    return model, tokenizer