






'''simple docstring'''

from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available, is_vision_available


SCREAMING_SNAKE_CASE__     =       {
    'configuration_mask2former': [
        'MASK2FORMER_PRETRAINED_CONFIG_ARCHIVE_MAP',
        'Mask2FormerConfig',
    ],
}

try:
     if not is_vision_available():
          raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
     pass
else:
     SCREAMING_SNAKE_CASE__     =       ['Mask2FormerImageProcessor']

try:
     if not is_torch_available():
          raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
     pass
else:
     SCREAMING_SNAKE_CASE__     =       [
         'MASK2FORMER_PRETRAINED_MODEL_ARCHIVE_LIST',
         'Mask2FormerForUniversalSegmentation',
         'Mask2FormerModel',
         'Mask2FormerPreTrainedModel',
     ]

if TYPE_CHECKING:
     from .configuration_maskaformer import MASK2FORMER_PRETRAINED_CONFIG_ARCHIVE_MAP, MaskaFormerConfig

     try:
          if not is_vision_available():
               raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
          pass
     else:
          from .image_processing_maskaformer import MaskaFormerImageProcessor

     try:
          if not is_torch_available():
               raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
          pass
     else:
          from .modeling_maskaformer import (
              MASK2FORMER_PRETRAINED_MODEL_ARCHIVE_LIST,
              MaskaFormerForUniversalSegmentation,
              MaskaFormerModel,
              MaskaFormerPreTrainedModel,
          )


else:
     import sys

     SCREAMING_SNAKE_CASE__     =       _LazyModule(__name__, globals()['__file__'], _import_structure)








'''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     a_						(      lowerCamelCase    ):
  lowercase							=				"""Salesforce/blip-image-captioning-base"""
  lowercase							=				(
      """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."""
  )
  lowercase							=				"""image_captioner"""
  lowercase							=				AutoModelForVisionaSeq

  lowercase							=				["""image"""]
  lowercase							=				["""text"""]

  def __init__(				self      ,		*_SCREAMING_SNAKE_CASE      ,		**_SCREAMING_SNAKE_CASE				)  ->				Any:



        """simple docstring"""






        requires_backends(self      ,		["""vision"""]				)
        super().__init__(*_SCREAMING_SNAKE_CASE      ,		**_SCREAMING_SNAKE_CASE				)

  def   A__						(				self      ,		_SCREAMING_SNAKE_CASE				)  ->				str:



        """simple docstring"""






        return self.pre_processor(images=_SCREAMING_SNAKE_CASE      ,		return_tensors="""pt"""				)

  def   A__						(				self      ,		_SCREAMING_SNAKE_CASE				)  ->				str:



        """simple docstring"""






        return self.model.generate(**_SCREAMING_SNAKE_CASE				)


  def   A__						(				self      ,		_SCREAMING_SNAKE_CASE				)  ->				Union[str, Any]:



        """simple docstring"""






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