import torchvision
import string

class Config:
    characters = list(string.digits + string.ascii_lowercase)
    character_lookup = {char: idx for idx, char in enumerate(characters)}
    character_lookup['<'] = max(character_lookup.values()) + 1
    untransform = torchvision.transforms.ToPILImage()
    device = 'cuda'
    def input_transform(im):
        T = torchvision.transforms.Compose([
                torchvision.transforms.Resize([28, 28])
            ])
        return T(im)
    
