


'''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

from ..models.auto import AutoModelForVisionaSeq
from ..utils import requires_backends
from .base import PipelineTool


if TYPE_CHECKING:
		from PIL import Image
class        lowercase     (							A__ ):

		"""simple docstring"""

		_a   			=    'Salesforce/blip-image-captioning-base'
		_a   			=    (
		    'This is a tool that generates a description of an image. It takes an input named `image` which should be the '
		    'image to caption, and returns a text that contains the description in English.'
		)
		_a   			=    'image_captioner'
		_a   			=    AutoModelForVisionaSeq

		_a   			=    ['image']
		_a   			=    ['text']




		def __init__(		self   ,       *UpperCamelCase_   ,       **UpperCamelCase_							):

							'''simple docstring'''


							requires_backends(self   ,       ['''vision''']							)
							super().__init__(*UpperCamelCase_   ,       **UpperCamelCase_							)




		def    lowerCAmelCase__    (		self   ,       UpperCamelCase_							):

							'''simple docstring'''


							return self.pre_processor(images=UpperCamelCase_   ,       return_tensors='''pt'''							)




		def    lowerCAmelCase__    (		self   ,       UpperCamelCase_							):

							'''simple docstring'''


							return self.model.generate(**UpperCamelCase_							)




		def    lowerCAmelCase__    (		self   ,       UpperCamelCase_							):

							'''simple docstring'''


							return self.pre_processor.batch_decode(UpperCamelCase_   ,       skip_special_tokens=UpperCamelCase_							)[0].strip()



'''simple docstring'''
# Copyright 2021 The HuggingFace 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 argparse

from .config import config_command_parser
from .config_args import default_config_file, load_config_from_file  # noqa: F401
from .default import default_command_parser
from .update import update_command_parser


def       a				(      __a=None       )  ->     List[str]:



					'''simple docstring'''






					UpperCamelCase__       :Optional[Any]							 =			argparse.ArgumentParser(add_help=__a			, allow_abbrev=__a       )
					# The main config parser
					UpperCamelCase__       :str							 =			config_command_parser(__a       )
					# The subparser to add commands to
					UpperCamelCase__       :Union[str, Any]							 =			config_parser.add_subparsers(title='''subcommands'''			, dest='''subcommand'''       )

					# Then add other parsers with the parent parser
					default_command_parser(__a			, parents=[parent_parser]       )
					update_command_parser(__a			, parents=[parent_parser]       )

					return config_parser


def       a				(      )  ->     Any:



					'''simple docstring'''






					UpperCamelCase__       :int							 =			get_config_parser()
					UpperCamelCase__       :List[Any]							 =			config_parser.parse_args()

					if not hasattr(__a			, '''func'''       ):
										config_parser.print_help()
										exit(1       )

					# Run
					args.func(__a       )


if __name__ == "__main__":
		main()