def __init__(self, opt, vocab, device):
self.opt = opt
self.device = device
print(self.device)
checkpoint = torch.load(opt.model)
model = Models.TaggerModel(opt, vocab)
model.load_state_dict(checkpoint['model'])
print('[Info] Trained model state loaded.')
self.model = model.to(self.device)
self.model.eval()
