U 9%eX@s8dZddlZddlmZddlmZGdddeZdS)z Processor class for TrOCR. N)contextmanager)ProcessorMixincsneZdZdZddgZdZdZdfdd Zd d Zd d Z d dZ e ddZ e ddZe ddZZS)TrOCRProcessora Constructs a TrOCR processor which wraps a vision image processor and a TrOCR tokenizer into a single processor. [`TrOCRProcessor`] offers all the functionalities of [`ViTImageProcessor`/`DeiTImageProcessor`] and [`RobertaTokenizer`/`XLMRobertaTokenizer`]. See the [`~TrOCRProcessor.__call__`] and [`~TrOCRProcessor.decode`] for more information. Args: image_processor ([`ViTImageProcessor`/`DeiTImageProcessor`]): An instance of [`ViTImageProcessor`/`DeiTImageProcessor`]. The image processor is a required input. tokenizer ([`RobertaTokenizer`/`XLMRobertaTokenizer`]): An instance of [`RobertaTokenizer`/`XLMRobertaTokenizer`]. The tokenizer is a required input. image_processor tokenizerZAutoImageProcessorZ AutoTokenizerNc srd}d|kr"tdt|d}|dk r.|n|}|dkrBtd|dkrRtdt|||j|_d|_ dS)Nfeature_extractorzhThe `feature_extractor` argument is deprecated and will be removed in v5, use `image_processor` instead.z)You need to specify an `image_processor`.z"You need to specify a `tokenizer`.F) warningswarn FutureWarningpop ValueErrorsuper__init__rcurrent_processor_in_target_context_manager)selfrrkwargsr __class__i/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/transformers/models/trocr/processing_trocr.pyr*s zTrOCRProcessor.__init__cOs|jr|j||S|dd}|dd}t|dkrJ|d}|dd}|dkrb|dkrbtd|dk r||j|f||}|dk r|j|f|}|dkr|S|dkr|S|d|d<|SdS) a When used in normal mode, this method forwards all its arguments to AutoImageProcessor's [`~AutoImageProcessor.__call__`] and returns its output. If used in the context [`~TrOCRProcessor.as_target_processor`] this method forwards all its arguments to TrOCRTokenizer's [`~TrOCRTokenizer.__call__`]. Please refer to the doctsring of the above two methods for more information. imagesNtextrzBYou need to specify either an `images` or `text` input to process.Z input_idslabels)rrr lenr rr)rargsrrrinputs encodingsrrr__call__>s&      zTrOCRProcessor.__call__cOs|jj||S)z This method forwards all its arguments to TrOCRTokenizer's [`~PreTrainedTokenizer.batch_decode`]. Please refer to the docstring of this method for more information. )r batch_decoderrrrrrr!_szTrOCRProcessor.batch_decodecOs|jj||S)z This method forwards all its arguments to TrOCRTokenizer's [`~PreTrainedTokenizer.decode`]. Please refer to the docstring of this method for more information. )rdecoder"rrrr#fszTrOCRProcessor.decodeccs0tdd|_|j|_dV|j|_d|_dS)z Temporarily sets the tokenizer for processing the input. Useful for encoding the labels when fine-tuning TrOCR. z`as_target_processor` is deprecated and will be removed in v5 of Transformers. You can process your labels by using the argument `text` of the regular `__call__` method (either in the same call as your images inputs, or in a separate call.TNF)r r rrrrrrrras_target_processormsz"TrOCRProcessor.as_target_processorcCstdt|jS)Nzg`feature_extractor_class` is deprecated and will be removed in v5. Use `image_processor_class` instead.)r r r image_processor_classr$rrrfeature_extractor_class}s z&TrOCRProcessor.feature_extractor_classcCstdt|jS)Nz[`feature_extractor` is deprecated and will be removed in v5. Use `image_processor` instead.)r r r rr$rrrrs z TrOCRProcessor.feature_extractor)NN)__name__ __module__ __qualname____doc__ attributesr&Ztokenizer_classrr r!r#rr%propertyr'r __classcell__rrrrrs !  r)r+r contextlibrZprocessing_utilsrrrrrrs