53 @@ +import torch +import re +from app.config import config +from app.trainer.train import load_model + +class Utterance_Rewriter: + def __init__(self, config): + + self.device = config.DEVICE + self.tokenizer, self.model = load_model(config) + self.model.eval() + + def ...