dvsth
/

LEGIT-TrOCR-MT

Feature Extraction

vision-encoder-decoder

Model card Files Files and versions Community

dvsth commited on Feb 28, 2023

Commit

99407d3

·

1 Parent(s): 838ec33

Upload model

Files changed (1) hide show

LegibilityModel.py +1 -1

LegibilityModel.py CHANGED Viewed

@@ -2,7 +2,7 @@ import torch.nn as nn
 from transformers import VisionEncoderDecoderModel, PreTrainedModel, AutoConfig
 class LegibilityModel(PreTrainedModel):
-    def __init__(self):
         config = AutoConfig.from_pretrained("microsoft/trocr-base-handwritten")
         super(LegibilityModel, self).__init__(config=config)

 from transformers import VisionEncoderDecoderModel, PreTrainedModel, AutoConfig
 class LegibilityModel(PreTrainedModel):
+    def __init__(self, config):
         config = AutoConfig.from_pretrained("microsoft/trocr-base-handwritten")
         super(LegibilityModel, self).__init__(config=config)