| 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)] | |