"""simple docstring"""





# 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.
from typing import TYPE_CHECKING

import torch

from ..models.auto import AutoModelForVisualQuestionAnswering, AutoProcessor
from ..utils import requires_backends
from .base import PipelineTool


if TYPE_CHECKING:
				from PIL import Image
class  _snake_case     (					a__		):
						snake_case__			  =			"dandelin/vilt-b32-finetuned-vqa"
						snake_case__			  =			(
						    "This is a tool that answers a question about an image. It takes an input named `image` which should be the "
						    "image containing the information, as well as a `question` which should be the question in English. It "
						    "returns a text that is the answer to the question."
						)
						snake_case__			  =			"image_qa"
						snake_case__			  =			AutoProcessor
						snake_case__			  =			AutoModelForVisualQuestionAnswering

						snake_case__			  =			["image", "text"]
						snake_case__			  =			["text"]






						def __init__(       self      :			List[Any]							,					*UpperCAmelCase      :			Any							,					**UpperCAmelCase      :			List[Any]    ):
													requires_backends(self							,					["vision"]    )
													super().__init__(*UpperCAmelCase							,					**UpperCAmelCase    )






						def       lowerCamelCase__     (       self      :			int							,					UpperCAmelCase      :			"Image"							,					UpperCAmelCase      :			str    ):
													return self.pre_processor(UpperCAmelCase							,					UpperCAmelCase							,					return_tensors="pt"    )






						def       lowerCamelCase__     (       self      :			int							,					UpperCAmelCase      :			Dict    ):
													with torch.no_grad():
																				return self.model(**UpperCAmelCase    ).logits







						def       lowerCamelCase__     (       self      :			Dict							,					UpperCAmelCase      :			int    ):
													__lowerCamelCase					:		List[str]		   =     outputs.argmax(-1    ).item()
													return self.model.config.idalabel[idx]
"""simple docstring"""





from typing import TYPE_CHECKING

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


__A  =				{
    '''configuration_bert''': ['''BERT_PRETRAINED_CONFIG_ARCHIVE_MAP''', '''BertConfig''', '''BertOnnxConfig'''],
    '''tokenization_bert''': ['''BasicTokenizer''', '''BertTokenizer''', '''WordpieceTokenizer'''],
}

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

try:
				if not is_torch_available():
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				pass
else:
				__A  =				[
				    '''BERT_PRETRAINED_MODEL_ARCHIVE_LIST''',
				    '''BertForMaskedLM''',
				    '''BertForMultipleChoice''',
				    '''BertForNextSentencePrediction''',
				    '''BertForPreTraining''',
				    '''BertForQuestionAnswering''',
				    '''BertForSequenceClassification''',
				    '''BertForTokenClassification''',
				    '''BertLayer''',
				    '''BertLMHeadModel''',
				    '''BertModel''',
				    '''BertPreTrainedModel''',
				    '''load_tf_weights_in_bert''',
				]

try:
				if not is_tf_available():
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				pass
else:
				__A  =				[
				    '''TF_BERT_PRETRAINED_MODEL_ARCHIVE_LIST''',
				    '''TFBertEmbeddings''',
				    '''TFBertForMaskedLM''',
				    '''TFBertForMultipleChoice''',
				    '''TFBertForNextSentencePrediction''',
				    '''TFBertForPreTraining''',
				    '''TFBertForQuestionAnswering''',
				    '''TFBertForSequenceClassification''',
				    '''TFBertForTokenClassification''',
				    '''TFBertLMHeadModel''',
				    '''TFBertMainLayer''',
				    '''TFBertModel''',
				    '''TFBertPreTrainedModel''',
				]
try:
				if not is_tensorflow_text_available():
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				pass
else:
				__A  =				['''TFBertTokenizer''']

try:
				if not is_flax_available():
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				pass
else:
				__A  =				[
				    '''FlaxBertForCausalLM''',
				    '''FlaxBertForMaskedLM''',
				    '''FlaxBertForMultipleChoice''',
				    '''FlaxBertForNextSentencePrediction''',
				    '''FlaxBertForPreTraining''',
				    '''FlaxBertForQuestionAnswering''',
				    '''FlaxBertForSequenceClassification''',
				    '''FlaxBertForTokenClassification''',
				    '''FlaxBertModel''',
				    '''FlaxBertPreTrainedModel''',
				]

if TYPE_CHECKING:
				from .configuration_bert import BERT_PRETRAINED_CONFIG_ARCHIVE_MAP, BertConfig, BertOnnxConfig
				from .tokenization_bert import BasicTokenizer, BertTokenizer, WordpieceTokenizer

				try:
								if not is_tokenizers_available():
												raise OptionalDependencyNotAvailable()
				except OptionalDependencyNotAvailable:
								pass
				else:
								from .tokenization_bert_fast import BertTokenizerFast

				try:
								if not is_torch_available():
												raise OptionalDependencyNotAvailable()
				except OptionalDependencyNotAvailable:
								pass
				else:
								from .modeling_bert import (
								    BERT_PRETRAINED_MODEL_ARCHIVE_LIST,
								    BertForMaskedLM,
								    BertForMultipleChoice,
								    BertForNextSentencePrediction,
								    BertForPreTraining,
								    BertForQuestionAnswering,
								    BertForSequenceClassification,
								    BertForTokenClassification,
								    BertLayer,
								    BertLMHeadModel,
								    BertModel,
								    BertPreTrainedModel,
								    load_tf_weights_in_bert,
								)

				try:
								if not is_tf_available():
												raise OptionalDependencyNotAvailable()
				except OptionalDependencyNotAvailable:
								pass
				else:
								from .modeling_tf_bert import (
								    TF_BERT_PRETRAINED_MODEL_ARCHIVE_LIST,
								    TFBertEmbeddings,
								    TFBertForMaskedLM,
								    TFBertForMultipleChoice,
								    TFBertForNextSentencePrediction,
								    TFBertForPreTraining,
								    TFBertForQuestionAnswering,
								    TFBertForSequenceClassification,
								    TFBertForTokenClassification,
								    TFBertLMHeadModel,
								    TFBertMainLayer,
								    TFBertModel,
								    TFBertPreTrainedModel,
								)

				try:
								if not is_tensorflow_text_available():
												raise OptionalDependencyNotAvailable()
				except OptionalDependencyNotAvailable:
								pass
				else:
								from .tokenization_bert_tf import TFBertTokenizer

				try:
								if not is_flax_available():
												raise OptionalDependencyNotAvailable()
				except OptionalDependencyNotAvailable:
								pass
				else:
								from .modeling_flax_bert import (
								    FlaxBertForCausalLM,
								    FlaxBertForMaskedLM,
								    FlaxBertForMultipleChoice,
								    FlaxBertForNextSentencePrediction,
								    FlaxBertForPreTraining,
								    FlaxBertForQuestionAnswering,
								    FlaxBertForSequenceClassification,
								    FlaxBertForTokenClassification,
								    FlaxBertModel,
								    FlaxBertPreTrainedModel,
								)

else:
				import sys

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