from transformers import PretrainedConfig

# Configuração do Modelo (similar ao PretrainedConfig do Hugging Face)
class CaptchaConfig(PretrainedConfig):
    model_type = "captcha"
    def __init__(self, input_dim=(40, 110), output_ndigits=5, output_vocab_size=10, vocab=None, **kwargs):
        super().__init__(**kwargs)
        self.input_dim = input_dim
        self.output_ndigits = output_ndigits
        self.output_vocab_size = output_vocab_size
        self.vocab = vocab if vocab else [str(i) for i in range(10)]