"""simple docstring"""




from typing import TYPE_CHECKING

from ...utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_flax_available,
    is_sentencepiece_available,
    is_tf_available,
    is_tokenizers_available,
    is_torch_available,
)


UpperCamelCase__    							= {
    '''configuration_xlm_roberta''': [
        '''XLM_ROBERTA_PRETRAINED_CONFIG_ARCHIVE_MAP''',
        '''XLMRobertaConfig''',
        '''XLMRobertaOnnxConfig''',
    ],
}

try:
      if not is_sentencepiece_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      UpperCamelCase__    							= ['''XLMRobertaTokenizer''']

try:
      if not is_tokenizers_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      UpperCamelCase__    							= ['''XLMRobertaTokenizerFast''']

try:
      if not is_torch_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      UpperCamelCase__    							= [
          '''XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST''',
          '''XLMRobertaForCausalLM''',
          '''XLMRobertaForMaskedLM''',
          '''XLMRobertaForMultipleChoice''',
          '''XLMRobertaForQuestionAnswering''',
          '''XLMRobertaForSequenceClassification''',
          '''XLMRobertaForTokenClassification''',
          '''XLMRobertaModel''',
          '''XLMRobertaPreTrainedModel''',
      ]

try:
      if not is_tf_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      UpperCamelCase__    							= [
          '''TF_XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST''',
          '''TFXLMRobertaForCausalLM''',
          '''TFXLMRobertaForMaskedLM''',
          '''TFXLMRobertaForMultipleChoice''',
          '''TFXLMRobertaForQuestionAnswering''',
          '''TFXLMRobertaForSequenceClassification''',
          '''TFXLMRobertaForTokenClassification''',
          '''TFXLMRobertaModel''',
          '''TFXLMRobertaPreTrainedModel''',
      ]

try:
      if not is_flax_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      UpperCamelCase__    							= [
          '''FLAX_XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST''',
          '''FlaxXLMRobertaForMaskedLM''',
          '''FlaxXLMRobertaForCausalLM''',
          '''FlaxXLMRobertaForMultipleChoice''',
          '''FlaxXLMRobertaForQuestionAnswering''',
          '''FlaxXLMRobertaForSequenceClassification''',
          '''FlaxXLMRobertaForTokenClassification''',
          '''FlaxXLMRobertaModel''',
          '''FlaxXLMRobertaPreTrainedModel''',
      ]

if TYPE_CHECKING:
      from .configuration_xlm_roberta import (
          XLM_ROBERTA_PRETRAINED_CONFIG_ARCHIVE_MAP,
          XLMRobertaConfig,
          XLMRobertaOnnxConfig,
      )

      try:
            if not is_sentencepiece_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .tokenization_xlm_roberta import XLMRobertaTokenizer

      try:
            if not is_tokenizers_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .tokenization_xlm_roberta_fast import XLMRobertaTokenizerFast

      try:
            if not is_torch_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .modeling_xlm_roberta import (
                XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST,
                XLMRobertaForCausalLM,
                XLMRobertaForMaskedLM,
                XLMRobertaForMultipleChoice,
                XLMRobertaForQuestionAnswering,
                XLMRobertaForSequenceClassification,
                XLMRobertaForTokenClassification,
                XLMRobertaModel,
                XLMRobertaPreTrainedModel,
            )

      try:
            if not is_tf_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .modeling_tf_xlm_roberta import (
                TF_XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST,
                TFXLMRobertaForCausalLM,
                TFXLMRobertaForMaskedLM,
                TFXLMRobertaForMultipleChoice,
                TFXLMRobertaForQuestionAnswering,
                TFXLMRobertaForSequenceClassification,
                TFXLMRobertaForTokenClassification,
                TFXLMRobertaModel,
                TFXLMRobertaPreTrainedModel,
            )

      try:
            if not is_flax_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .modeling_flax_xlm_roberta import (
                FLAX_XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST,
                FlaxXLMRobertaForCausalLM,
                FlaxXLMRobertaForMaskedLM,
                FlaxXLMRobertaForMultipleChoice,
                FlaxXLMRobertaForQuestionAnswering,
                FlaxXLMRobertaForSequenceClassification,
                FlaxXLMRobertaForTokenClassification,
                FlaxXLMRobertaModel,
                FlaxXLMRobertaPreTrainedModel,
            )

else:
      import sys

      UpperCamelCase__    							= _LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)




# Copyright 2023 The HuggingFace Inc. team. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import torch

from ..models.auto import AutoModelForSequenceClassification, AutoTokenizer
from .base import PipelineTool





class 					A							(    UpperCAmelCase_					):
   __UpperCAmelCase			:			List[Any]              =      'facebook/bart-large-mnli'
   __UpperCAmelCase			:			Optional[Any]              =      (
       'This is a tool that classifies an English text using provided labels. It takes two inputs: `text`, which '
       'should be the text to classify, and `labels`, which should be the list of labels to use for classification. '
       'It returns the most likely label in the list of provided `labels` for the input text.'
   )
   __UpperCAmelCase			:			Optional[int]              =      'text_classifier'
   __UpperCAmelCase			:			int              =      AutoTokenizer
   __UpperCAmelCase			:			Dict              =      AutoModelForSequenceClassification

   __UpperCAmelCase			:			int              =      ['text', ['text']]
   __UpperCAmelCase			:			Optional[int]              =      ['text']


   def       lowercase_				(self					:						List[Any]		)						->    List[str]:
         """simple docstring"""



         super().setup()
         UpperCAmelCase__								=  self.model.config
         UpperCAmelCase__								=  -1
         for idx, label in config.idalabel.items():
               if label.lower().startswith("entail"		):
                     UpperCAmelCase__								=  int(__UpperCAmelCase		)
         if self.entailment_id == -1:
               raise ValueError("Could not determine the entailment ID from the model config, please pass it at init."		)


   def       lowercase_				(self					:						Union[str, Any]	,  __UpperCAmelCase					:						Dict	,  __UpperCAmelCase					:						int		)						->    Optional[int]:
         """simple docstring"""



         UpperCAmelCase__								=  labels
         return self.pre_processor(
             [text] * len(__UpperCAmelCase		)	,  [f"""This example is {label}""" for label in labels]	,  return_tensors="pt"	,  padding="max_length"	,  )





   def       lowercase_				(self					:						Dict	,  __UpperCAmelCase					:						Tuple		)						->    int:
         """simple docstring"""



         UpperCAmelCase__								=  outputs.logits
         UpperCAmelCase__								=  torch.argmax(logits[:, 2]		).item()
         return self._labels[label_id]



