U ʹj @sddlZddlmZmZmZmZmZddlmZddl m Z ddl m Z m Z mZmZmZddlmZeeZdd d d d d ddddddddddddddddddd gZGd!d"d"ZdS)#N)AnyDictListOptionalUnion) constants)HfApi)build_hf_headers get_sessionis_pillow_availableloggingvalidate_hf_hub_args)_deprecate_methodztext-classificationztoken-classificationztable-question-answeringzquestion-answeringzzero-shot-classification translationZ summarizationZconversationalzfeature-extractionztext-generationztext2text-generationz fill-maskzsentence-similarityztext-to-speechzautomatic-speech-recognitionzaudio-to-audiozaudio-classificationzvoice-activity-detectionzimage-classificationzobject-detectionzimage-segmentationz text-to-imagezimage-to-imageztabular-classificationztabular-regressionc @seZdZdZeeddddeeeeeeddd Z d d Z dee ee e ee e efee eeeed d dZdS) InferenceApiaClient to configure requests and make calls to the HuggingFace Inference API. Example: ```python >>> from huggingface_hub.inference_api import InferenceApi >>> # Mask-fill example >>> inference = InferenceApi("bert-base-uncased") >>> inference(inputs="The goal of life is [MASK].") [{'sequence': 'the goal of life is life.', 'score': 0.10933292657136917, 'token': 2166, 'token_str': 'life'}] >>> # Question Answering example >>> inference = InferenceApi("deepset/roberta-base-squad2") >>> inputs = { ... "question": "What's my name?", ... "context": "My name is Clara and I live in Berkeley.", ... } >>> inference(inputs) {'score': 0.9326569437980652, 'start': 11, 'end': 16, 'answer': 'Clara'} >>> # Zero-shot example >>> inference = InferenceApi("typeform/distilbert-base-uncased-mnli") >>> inputs = "Hi, I recently bought a device from your company but it is not working as advertised and I would like to get reimbursed!" >>> params = {"candidate_labels": ["refund", "legal", "faq"]} >>> inference(inputs, params) {'sequence': 'Hi, I recently bought a device from your company but it is not working as advertised and I would like to get reimbursed!', 'labels': ['refund', 'faq', 'legal'], 'scores': [0.9378499388694763, 0.04914155602455139, 0.013008488342165947]} >>> # Overriding configured task >>> inference = InferenceApi("bert-base-uncased", task="feature-extraction") >>> # Text-to-image >>> inference = InferenceApi("stabilityai/stable-diffusion-2-1") >>> inference("cat") >>> # Return as raw response to parse the output yourself >>> inference = InferenceApi("mio/amadeus") >>> response = inference("hello world", raw_response=True) >>> response.headers {"Content-Type": "audio/flac", ...} >>> response.content # raw bytes from server b'(...)' ``` z1.0z`InferenceApi` client is deprecated in favor of the more feature-complete `InferenceClient`. Check out this guide to learn how to convert your script to use it: https://huggingface.co/docs/huggingface_hub/guides/inference#legacy-inferenceapi-client.)versionmessageNF)repo_idtasktokengpucCsd|d|_t|d|_t|dj|d}|js<|sr#r#r#r$r.s6. 4r)r7typingrrrrrr/rZhf_apir utilsr r r r rZutils._deprecationrZ get_loggerr?rrrr#r#r#r$sB    !