


"""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 ..models.whisper import WhisperForConditionalGeneration, WhisperProcessor
from .base import PipelineTool
class 					__magic_name__		(			UpperCamelCase_     ):
							__A			:      Optional[Any]        =   "openai/whisper-base"
							__A			:      List[Any]        =   (
							    "This is a tool that transcribes an audio into text. It takes an input named `audio` and returns the "
							    "transcribed text."
							)
							__A			:      Any        =   "transcriber"
							__A			:      List[str]        =   WhisperProcessor
							__A			:      int        =   WhisperForConditionalGeneration

							__A			:      Tuple        =   ["audio"]
							__A			:      Union[str, Any]        =   ["text"]






							def  __snake_case						(   self  :    Optional[Any]  ,     snake_case__  :    Any				):

														'''simple docstring'''






														return self.pre_processor(UpperCamelCase__  ,     return_tensors='''pt'''				).input_features






							def  __snake_case						(   self  :    Any  ,     snake_case__  :    int				):

														'''simple docstring'''






														return self.model.generate(inputs=UpperCamelCase__				)







							def  __snake_case						(   self  :    Any  ,     snake_case__  :    Optional[int]				):

														'''simple docstring'''






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





"""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.
import re

from ..utils import cached_file


# docstyle-ignore
UpperCAmelCase      		=			'''
Human: <<task>>

Assistant: '''


UpperCAmelCase      		=			'''huggingface-tools/default-prompts'''
UpperCAmelCase      		=			{'''chat''': '''chat_prompt_template.txt''', '''run''': '''run_prompt_template.txt'''}






def       lowerCamelCase					(a_						:int				,				a_						:str				,				a_						:Dict="run")     ->							Optional[Any]:
       if prompt_or_repo_id is None:
              lowercase		:Tuple       							=				DEFAULT_PROMPTS_REPO

       # prompt is considered a repo ID when it does not contain any kind of space
       if re.search('''\\s'''				,				a_) is not None:
              return prompt_or_repo_id

       lowercase		:List[str]       							=				cached_file(
           a_				,				PROMPT_FILES[mode]				,				repo_type='''dataset'''				,				user_agent={'''agent''': agent_name})
       with open(a_				,				'''r'''				,				encoding='''utf-8''') as f:
              return f.read()

