modifica modeling_tjmg

modeling_tjmg.py

@@ -5,18 +5,6 @@ from transformers import PreTrainedModel
 from torchvision import transforms
 from PIL import Image
 
-from transformers import PretrainedConfig
-
-# Configuração do Modelo (similar ao PretrainedConfig do Hugging Face)
-class CaptchaConfig(PretrainedConfig):
-    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)]
-
-
 # Modelo baseado na arquitetura do CaptchaCNN
 class CaptchaModel(PreTrainedModel):
     config_class = CaptchaConfig